sklearn回归简单示例

1、线性回归:

import pandas as pd
import numpy as np
from sklearn import linear_model as lm

#准备数据，fit函数要求x项为矩阵，y项为序列，故只有单个自变量时需要转置
a=pd.read_excel(r'D:\baidu\Desktop\1.xls')
b=a.icol(1)
b=[[x] for x in b]#或者b=b.reshape(len(b),1)
c=a.icol(2)

#训练模型
f=lm.LinearRegression()
f1=f.fit(b,c)

#获得结果
c,i,p=f1.coef_,f1.intercept_,f1.predict(100)#f1.coef_为系数，intercept_为常数项,predict()获得预测结果
print(c,i,p)