# 示例代码:一个简单的线性回归模型使用Python的scikit-learn库
# 导入必要的库
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error
import numpy as np
# 生成一些示例数据
# 假设我们有一个简单的线性关系 y = 2x + 1 + 一些噪声
X = np.random.rand(100, 1) * 10 # 100个样本,1个特征
y = 2 * X.squeeze() + 1 + np.random.randn(100) * 2 # 添加一些噪声
# 将数据分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建线性回归模型
model = LinearRegression()
# 训练模型
model.fit(X_train, y_train)
# 使用模型进行预测
y_pred = model.predict(X_test)
# 评估模型性能
mse = mean_squared_error(y_test, y_pred)
print(f"Mean Squared Error: {mse}")
# 输出模型的系数和截距
print(f"Coefficients: {model.coef_}")
print(f"Intercept: {model.intercept_}")
train_test_split用于划分训练集和测试集,LinearRegression用于创建线性回归模型,mean_squared_error用于评估模型性能,以及numpy用于生成示例数据。LinearRegression创建一个线性回归模型,并用训练数据对其进行训练。上一篇:python求余数的方法
下一篇:编程python语言
Laravel PHP 深圳智简公司。版权所有©2023-2043 LaravelPHP 粤ICP备2021048745号-3
Laravel 中文站