Jac*_*ova 12 machine-learning linear-regression scikit-learn
我正在尝试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()
Run Code Online (Sandbox Code Playgroud)
我的问题是:我希望这个线性模型有2个系数:截距项和x系数,我怎么得到一个?

Jac*_*ova 20
哎呀
我没注意到拦截是模型的一个独立属性!
print('Intercept: \n', model.intercept_)
Run Code Online (Sandbox Code Playgroud)
看看这里的文档
intercept_:array
线性模型中的独立项.
| 归档时间: |
|
| 查看次数: |
9167 次 |
| 最近记录: |