在 Python 中编写最佳拟合线函数

Question

我将 python 和 scikitlearn 用于线性回归和可视化目的。

我需要创建自己的函数来返回渐变和最佳拟合线 y 截距，而不是使用来自 sklearn 的 linear_model.LinearRegression()。

m = (μ(x) * μ(y) - μ(x * y))/((μ(x))2 - μ(x2))

b = μ(y) - m * μ(x)

其中 μ 是平均函数

我正在努力在 python 中做到这一点。

任何帮助将不胜感激。

import matplotlib.pyplot as plt
import numpy as np
from sklearn import datasets, linear_model


diabetes = datasets.load_diabetes()



diabetes_X = diabetes.data[:, np.newaxis, 2]


diabetes_X_train = diabetes_X[:-20]
diabetes_X_test = diabetes_X[-20:]


diabetes_y_train = diabetes.target[:-20]
diabetes_y_test = diabetes.target[-20:]


regr = linear_model.LinearRegression()


regr.fit(diabetes_X_train, diabetes_y_train)


print('Coefficients: \n', regr.coef_)

print("Mean squared error: %.2f"
      % np.mean((regr.predict(diabetes_X_test) - diabetes_y_test) ** 2))

print('Variance score: %.2f' % regr.score(diabetes_X_test, diabetes_y_test))


plt.scatter(diabetes_X_test,  diabetes_y_test,  color='green', label="Test")
plt.scatter(diabetes_X_train,  diabetes_y_train,  color='red', label ='Training')
plt.plot(diabetes_X_test, regr.predict(diabetes_X_test), color='blue',
     linewidth=3, label='Best fit line')

plt.xticks(())
plt.yticks(())
plt.legend()

plt.show()

Answer 1

你可以使用 numpy 来计算平均值...

import numpy as np

m = (np.mean(x) * np.mean(y) − np.mean(x * y))/((np.mean(x))**2 − np.mean(x**2))
b = np.mean(y) − m * np.mean(x)

但是发送一些你已经完成的代码以获得更具体的帮助！