目录
使用的算法包括线性回归(LinearRegressjon)、岭回归(RldgeRegression)、LASSO(LeastAbsoluteShrinkageandSelectlonOperator)回归、决策树回归(DeclsjonTreeRegressor)、梯度提升树回归(Grad1e11tBoostmgDecisonTreeRegressor等。通过查阅资料熟悉如何使用sklearn调用。
线性回归模型:
fromsklearn.model_selectionimporttrain_test_split
fromsklearn.linear_modelimportLinearRegression
x_train,x_test,y_train,y_test=train_test_split(a,b,test_size=0.2)
clf=LinearRegression()
rf=clf.fit(x_train,y_train.ravel())
y_pred=rf.predict(x_test)
- KNN回归模型
fromsklearn.neighborsimportKNeighborsRegressor
x_train,x_test,y_train,y_test=train_test_split(a,b,test_size=0.2)
clf=KNeighborsRegressor()
rf=clf.fit(x_train,y_train.ravel())
y_pred=rf.predict(x_test)
- SVM回归模型
fromsklearn.svmimportSVR
x_train,x_test,y_train,y_test=train_test_split(a,b,test_size=0.2)
clf=SVR()
rf=clf.fit(x_train,y_train.ravel())
y_pred=rf.predict(x_test)
print("SVR结果如下:")
print("训练集分数:",rf.score(x_train,y_train))
print("验证集分数:",rf.score(x_test,y_test))
- 岭回归模型
fromsklearn.linear_modelimportRidge
x_train,x_test,y_train,y_test=train_test_split(a,b,test_size=0.2)
clf=Ridge()
rf=clf.fit(x_train,y_train.ravel())
y_pred=rf.predict(x_test)
print("Ridge结果如下:")
print("训练集分数:",rf.score(x_train,y_train))
print("验证集分数:",rf.score(x_test,y_test))