sklearn回归

线性回归:sklearn.linear_model.LinearRegression()

linear_regression = LinearRegression()  # 实例化
linear_regression.fit(data, label)  # 拟合
label_test = linear_regression.predict(data_test)  # 预测

线性支持向量机回归:sklearn.svm.LinearSVR()

linear_svr = LinearSVR(max_iter=7000)  # 实例化,设置最大迭代次数
linear_svr.fit(data, label)  # 拟合
label_test = linear_svr.predict(data_test)  # 预测

备注:
如果拟合时提示:
ConvergenceWarning: Liblinear failed to converge, increase the number of iterations.
(收敛警告:Liblinear无法收敛,增加了迭代次数。)
解决方法:手动设置max_iter提高迭代次数。

猜你喜欢

转载自blog.csdn.net/weixin_43793472/article/details/84479155