python 读取pmml文件 计算auc

#gbdt模型效果
from sklearn.metrics import roc_auc_score
from pypmml import Model

cols = ["start_distance","price","time_diff" ]
model = Model.fromFile('look_order_cross_city_gbdt_diff-2-14.pmml')#加载模型文件
result = model.predict((dfTest[cols]))
auc = roc_auc_score(dfTest['success'],result['probability(1)'])
print('gbdt auc',auc)