单项式模型
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
%matplotlib inline
#导入训练集和测试集
train_data = pd.read_csv("datas/T-R-train.csv")
test_data = pd.read_csv("datas/T-R-test.csv")
image.png
# 定义x和y
x_train = train_data["T"]
y_train = train_data["rate"]
x_test = test_data["T"]
y_test = test_data["rate"]
image.png
#将数据转换成一维
x_train = np.array(x_train).reshape(-1, 1)
x_test = np.array(x_test).reshape(-1, 1)
# 模型训练
line_model = LinearRegression()
line_model.fit(x_train, y_train)
#预测
y_train_predict = line_model.predict(x_train)
y_test_predict = line_model.predict(x_test)
# 查看模型的r2分数
from sklearn.metrics import r2_score
r2_train = r2_score(y_train, y_train_predict)
r2_test = r2_score(y_test, y_test_predict)
print(r2_train, r2_test) 0.016665703886981964 -0.758336343735132
#生成新的数据点
x_range = np.linspace(40, 90, 300).reshape(-1, 1)
new_y_predict = line_model.predict(x_range)
image.png
多项式模型——二阶
# 多项式模型 将原数据转换成二维
from sklearn.preprocessing import PolynomialFeatures
poly = PolynomialFeatures()
x_2_train = poly.fit_transform(x_train)
x_2_test = poly.transform(x_test)
# 模型训练
line_model2 = LinearRegression()
line_model2.fit(x_2_train, y_train)
#预测
y2_train_predict = line_model2.predict(x_2_train)
y2_test_predict = line_model2.predict(x_2_test)
#r2分数
r2_train2 = r2_score(y_train, y2_train_predict)
r2_test2 = r2_score(y_test, y2_test_predict)
print(r2_train2, r2_test2) 0.970051540068942 0.9963954556468684
#生成新的数据点
x2_range = np.linspace(40, 90, 300).reshape(-1, 1)
x2_range = poly.transform(x2_range)
new_y2_predict = line_model2.predict(x2_range)
image.png
多项式模型——五阶
#五阶模型
poly5 = PolynomialFeatures(degree=5)
x_5_train = poly5.fit_transform(x_train)
x_5_test = poly5.transform(x_test)
# 模型训练
line_model5 = LinearRegression()
line_model5.fit(x_5_train, y_train)
#预测
y5_train_predict = line_model5.predict(x_5_train)
y5_test_predict = line_model5.predict(x_5_test)
#r2分数
r2_train5 = r2_score(y_train, y5_train_predict)
r2_test5 = r2_score(y_test, y5_test_predict)
print(r2_train5, r2_test5)
#生成新的数据点
x5_range = np.linspace(40, 90, 300).reshape(-1, 1)
x5_range = poly5.transform(x5_range)
new_y5_predict = line_model5.predict(x5_range) 0.9978527267187658 0.5437837627379174
image.png
