fixtures.py 文件源码

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

项目:cesium_web 作者: cesium-ml 项目源码 文件源码
def add_file(model, create, value, *args, **kwargs):
        model_params = {
            "RandomForestClassifier": {
                "bootstrap": True, "criterion": "gini",
                "oob_score": False, "max_features": "auto",
                "n_estimators": 10, "random_state": 0},
            "RandomForestRegressor": {
                "bootstrap": True, "criterion": "mse",
                "oob_score": False, "max_features": "auto",
                "n_estimators": 10},
            "LinearSGDClassifier": {
                "loss": "hinge"},
            "LinearRegressor": {
                "fit_intercept": True}}
        fset_data, data = featurize.load_featureset(model.featureset.file_uri)
        model_data = MODELS_TYPE_DICT[model.type](**model_params[model.type])
        model_data.fit(fset_data, data['labels'])
        model.file_uri = pjoin('/tmp/', '{}.pkl'.format(str(uuid.uuid4())))
        joblib.dump(model_data, model.file_uri)
        DBSession().commit()
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号