def train_model_group(model_info,function='logistic'):
model = joblib.load(getTheFile('models/{}.m').format(function))
base_info = model_config.model_dim
user_id = model_info['user_id']
del model_info['user_id']
for key,value in base_info.items():
if key not in model_info.columns:
model_info[key] = value
#model_info = transform_data.transform_with_woe(model_info)
model_info = model_info[sorted(model_info.columns)]
x = np.matrix(model_info)
yt = model.predict(x)
ytp = model.predict_proba(x)
return user_id,yt,ytp
评论列表
文章目录