我正在关注this linear regression tutorial。这是我的代码:
import pandas as pd
from sklearn import linear_model
import matplotlib.pyplot as plt
dataframe = pd.read_fwf('brain_body.txt')
x_values = dataframe[['Brain']]
y_values = dataframe[['Body']]
body_reg = linear_model.LinearRegression()
body_reg.fit(x_values, y_values)
plt.scatter(x_values, y_values)
plt.plot(x_values, body_reg.predict(x_values))
plt.show()
当我运行脚本时,我没有错误,但图表似乎没有考虑到y值。我将数据点减少到三个,因此更容易看到:
我尝试用plt.ylim([-1000,7000])
手动更改y轴,但没有运气。
感谢您的任何建议!