sklearn回归简单示例

1、线性回归:

import pandas as pdimport numpy as npfrom 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)