from sklearn import linear_model
reg = linear_model.LinearRegression()
reg.fit(feature_train, target_train)
pred =reg.predict(feature_test)
print(reg.coef_) ##斜率
print(reg.intercept_) ##截距
print(reg.score(feature_train,target_train)) ##R平方
print(reg.score(feature_test, target_test))
##sklearn 中的年龄/净值回归
def studentReg(ages_train, net_worths_train):
from sklearn import linear_model
reg = linear_model.LinearRegression()
reg.fit(ages_train, net_worths_train)
return reg
##现在你练习提取信息
km_net_worth = reg.predict([[27]])
slope = reg.coef_
intercept = reg.intercept_
test_score = reg.score(ages_train, net_worths_train)
training_score = reg.score(ages_test, net_worths_test)
##提取斜率和截距
slope = reg.coef_
intercept = reg.intercept_
##回归分数:测试数据
test_score = reg.score(ages_train, net_worths_train)
training_score = reg.score(ages_test, net_worths_test)