#正则化 防止过度拟合
from sklearn.datasets import load_boston
from sklearn.linear_model import LinearRegression,SGDRegressor#小,大
from sklearn.preprocessing import StandardScaler
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_absolute_error,mean_squared_error
from sklearn.linear_model import Ridge
boston=load_boston()
features=boston.data
#线性模型对特征值敏感,标准化
scaler=StandardScaler()
features=scaler.fit_transform(features)
targets=boston.target
#划分数据集
X_train,X_test,y_train,y_test=train_test_split(features,targets,test_size=0.25)
ridge=Ridge(alpha=10)#注意调参数
ridge.fit(X_train,y_train)
y_predict=ridge.predict(X_test)
print("mean_absolute_error:",mean_absolute_error(y_test,y_predict))#平均误差
print("mean_squared_error:",mean_squared_error(y_test,y_predict))#平均方差
正则化和岭回归
最新推荐文章于 2022-11-25 20:13:57 发布