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()
评论列表
文章目录