简单的机器学习任务不起作用

Question

按照 Siraj 的演示：这是我的一行最适合单变量数据的代码

（文件可在here找到）：

#import dependencies
import pandas as pd
from sklearn import linear_model
import matplotlib.pyplot as plt

#read data
dataframe = pd.read_fwf('brain_body.txt')
x_values = dataframe[['Brain']]
y_values = dataframe[['Body']]

#train model on data
body_reg = linear_model.LinearRegression()
body_reg.fit(x_values, y_values)

#visualize results
plt.scatter(x_values, y_values)
plt.plot(x_values, body_reg.predict(x_values))
plt.show()

我总是弄得一团糟：

Answer 1

我的看起来一模一样。来这里看看是否有人发布了一些帮助。问题似乎与 2 个异常值（2547-4603 和 6654-5712）有关，因为它们明显更高。

Answer 2

Darren O'Rourke 的回答归功于他

由于 pandas 库中似乎存在差异，numpy 为我解决了这个问题。

import numpy as np
x_values = np.array(x_values)
y_values = np.array(y_values)

我希望这能解决问题。

干杯

简单的机器学习任务不起作用

Simple machine learning task not working

python

gradient

regression