run_model.py 文件源码

python
阅读 23 收藏 0 点赞 0 评论 0

项目:Mmodel 作者: gxrtbf 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号