第二章.线性回归以及非线性回归
import numpy as np
from numpy import genfromtxt
from sklearn import linear_model# 导入数据
data = genfromtxt('D:\\Data\\longley.csv', delimiter=',')# 数据切片
x_data = data[1:, 2:]
y_data = data[1:, 1]# 加载模型
model = linear_model.ElasticNetCV()# 拟合线性模型
model.fit(x_data, y_data)# 弹性网系数
coeff = model.alpha_
print('弹性网系数:', coeff)# 相关系数
rcoeff = model.coef_
print('相关系数:', rcoeff)# 预测值
pred = model.predict(x_data[2, np.newaxis])
print('第二行的预测数据:', pred, '实际数据:88.2')
①.数据