要使用Sklearn库实现线性回归,可以按照以下步骤进行操作:
- 导入所需的库和模块。
- 准备数据集。
- 划分训练集和测试集。
- 创建线性回归模型并设置参数。
- 训练模型并进行预测。
- 评估模型性能。
- 可视化结果。
以下是具体的代码实现:
# 导入所需的库和模块
import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error, r2_score
# 准备数据集
X = np.array([1, 2, 3, 4, 5]).reshape(-1, 1)
y = np.array([2, 4, 6, 8, 10])
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建线性回归模型并设置参数
lr = LinearRegression()
# 训练模型并进行预测
lr.fit(X_train, y_train)
y_pred = lr.predict(X_test)
# 评估模型性能
mse = mean_squared_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)
print("Mean Squared Error:", mse)
print("R2 Score:", r2)
# 可视化结果
plt.scatter(X_test, y_test, color='blue', label='Actual')
plt.plot(X_test, y_pred, color='red', label='Predicted')
plt.legend()
plt.show()
在这个例子中,我们首先导入了所需的库和模块,然后创建了一个简单的数据集。接下来,我们将数据集划分为训练集和测试集,然后创建了一个线性回归模型。我们使用训练集对模型进行训练,并使用测试集进行预测。最后,我们评估了模型的性能并可视化了结果。