[python] add predict the number