线性回归: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提高迭代次数。