【发布时间】:2017-03-26 22:20:08
【问题描述】:
我在一个包含 3756 行的数据集上训练了一个 SGDRegressor。这返回了我想要使用的 alpha 值。现在我想根据从 Javascript 前端接收到的新值集输出预测。这包括需要一种热编码的分类数据。我想知道如何将值传递给经过训练的回归模块并根据它生成预测。
def rmse_cv(model):
rmse = np.sqrt(-cross_val_score(model, X_train, y, scoring="neg_mean_squared_error", cv=5))
return(rmse)
alphas = [0.00005, 0.00015, 0.00045, 0.0135]
mv_sgd = [rmse_cv(SGDRegressor(alpha = alpha)).mean()
for alpha in alphas]
【问题讨论】:
标签: javascript python pandas machine-learning scikit-learn