SK-Learn随机梯度下降法

今天复习了梯度，感触多深，但我还是选择了sklearn哈哈

from sklearn.linear_model import SGDRegressor    # 梯度下降
from sklearn.datasets import load_boston
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler

导入数据集

sgd_reg=SGDRegressor()

X=load_boston().data
y=load_boston().target

X=X[y<50.0]
y=y[y<50.0]

X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.1,random_state=33)

标准化

standardScaler=StandardScaler()
standardScaler.fit(X_train)
X_train_stad=standardScaler.transform(X_train)
X_test_stad=standardScaler.transform(X_test)

sgd_reg.fit(X_train_stad,y_train)
sgd_reg.score(X_test_stad,y_test)

n_iter 表示你要重复对训练集训练多少遍
比较 n_iter 这个值的数量对结果影响

sgd_reg=SGDRegressor(n_iter=10)
sgd_reg.fit(X_train_stad,y_train)
sgd_reg.score(X_test_stad,y_test)