我尝试在一个简单的数据集上学习scikit-learn LinearRegression模型(来自Andrew NG coursera课程,我并不重要,请查看该图以供参考)
这是我的剧本
import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
dataset = np.loadtxt('../mlclass-ex1-008/mlclass-ex1/ex1data1.txt', delimiter=',')
X = dataset[:, 0]
Y = dataset[:, 1]
plt.figure()
plt.ylabel('Profit in $10,000s')
plt.xlabel('Population of City in 10,000s')
plt.grid()
plt.plot(X, Y, 'rx')
model = LinearRegression()
model.fit(X[:, np.newaxis], Y)
plt.plot(X, model.predict(X[:, np.newaxis]), color='blue', linewidth=3)
print('Coefficients: \n', model.coef_)
plt.show()
我的问题是: 我希望这个线性模型有2个系数:截距项和x系数,我怎么得到一个呢?
答案 0 :(得分:18)