[优达 机器学习入门]课程7：回归

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)