1. 程式人生 > >sklearn迴歸

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提高迭代次數。