【发布时间】:2017-12-21 20:22:12
【问题描述】:
按照 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()
我总是弄得一团糟:
【问题讨论】:
标签: python regression gradient