sklearn LinearRegression,为什么模型只返回一个系数?

时间:2015-04-17 15:10:36

标签: machine-learning scikit-learn linear-regression

我尝试在一个简单的数据集上学习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系数,我怎么得到一个呢?

enter image description here

1 个答案:

答案 0 :(得分:18)

哎呀

我没有注意到拦截是模型的独立属性!

print('Intercept: \n', model.intercept_)

查看文档here

  

intercept_:array

     

线性模型中的独立术语。