用ols构建最小二乘法进行线性回归模型训练后,如何利用模型进行预测?

其他的机器学习模型都可以使用predict进行预测,ols不可以,请问需要怎样使用ols模型?

请先 登录 后评论

1 个回答

Hao Jia

ols函数会根据传入的mode参数返回不同的结果,如果只需要进行预测功能,不关心模型的统计结果,在调用ols时选择mode=0即可。返回后是一个系数矩阵,model[0]表示截距(如果intercept=true)。

可将需要预测的值处理成矩阵,使用flatten(dot(matrix(X),model[1:]))+model[0]可以得到最后的模型预测结果。

请先 登录 后评论