python预测模型类型