【发布时间】:2021-01-04 19:38:50
【问题描述】:
我一直在尝试使用手动代码按照 Sklearn 线性回归库复制成本的结果。两者之间存在巨大差异,我无法弄清楚为什么。这是 Sklearn 的代码:
SkLearn 实施:
X_train, X_test, Y_train, Y_test = model_selection.train_test_split(X, Y, test_size=0.30)
classifier = sklearn.linear_model.LinearRegression()
classifier.fit(X_train,Y_train)
cost = np.sqrt(np.sum((np.dot(X_train,classifier.coef_.reshape(9,1)) + classifier.intercept_ - Y_train.reshape(478,1))**2))
print(cost)
cost = 4.236441942240197
我复制结果的尝试:
a = X_train_rev.shape
assert(X_train_rev.shape == (478,10)) # assert shape of the X_train_rev
Y_train = Y_train.reshape(478,1)
alpha = 0.0005 # Learning_Rate
coefficient = np.random.randn(1,10) # Initialisation of coefficients including intercept
# Loop through iterations
for i in range(100000):
cost = np.sqrt(np.sum((np.dot(X_train_rev,coefficient.T) - Y_train)**2)) # cost result
if i % 10000 == 0: print(cost)
grad = np.dot((np.dot(X_train_rev,coefficient.T) - Y_train).T, X_train_rev) # Compute Gradients
coefficient = coefficient - (alpha * grad) # adjust coefficients including intercept
Cost after Iterations:
45.23042864973579
10.428401916963285
10.428401916963285
10.428401916963285
10.428401916963285
10.428401916963285
10.428401916963285
10.428401916963285
10.428401916963285
10.428401916963285
根据我的手动代码,成本并没有进一步降低,而且与 Sklearn 的成本相差甚远。我尝试使用 alpha 变量,但 alpha 的任何增加都会导致成本趋于正无穷大。
请注意,我的手动代码中使用的 X_train_rev 数据在 Sklearn 训练集中有 10 个列/特征而不是 9 个特征,因为我在训练集中添加了一列“ones”来表示截距。同样,系数向量也包含截距。
【问题讨论】:
标签: python machine-learning scikit-learn linear-regression