机器学习:python 实现一个linear regression
- 行业动态
- 2024-07-14
- 1
要实现一个简单的线性回归模型,可以使用Python的sklearn库。首先安装库,然后导入所需模块,创建数据集,划分训练集和测试集,创建线性回归模型并训练,最后进行预测和评估。,,“ python,# 导入所需模块,import numpy as np,from sklearn.model_selection import train_test_split,from sklearn.linear_model import LinearRegression,from sklearn.metrics import mean_squared_error,,# 创建数据集,X = np.random.rand(100, 1),y = 2 * X + 1 + 0.1 * np.random.randn(100, 1),,# 划分训练集和测试集,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("均方误差:", mse),“
线性回归是一种预测模型,用于解决回归问题,它假设目标值与特征之间存在线性关系,下面是使用Python实现线性回归的步骤:
1、导入所需库
2、准备数据集
3、划分训练集和测试集
4、创建线性回归模型
5、训练模型
6、评估模型
7、使用模型进行预测
下面是具体的代码实现:
1. 导入所需库 import numpy as np import pandas as pd from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression from sklearn.metrics import mean_squared_error 2. 准备数据集 data = pd.read_csv('your_data.csv') # 请替换为你的数据文件路径 X = data[['feature1', 'feature2', 'feature3']] # 请替换为你的特征列名 y = data['target'] # 请替换为你的目标列名 3. 划分训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) 4. 创建线性回归模型 model = LinearRegression() 5. 训练模型 model.fit(X_train, y_train) 6. 评估模型 y_pred = model.predict(X_test) mse = mean_squared_error(y_test, y_pred) print('Mean Squared Error:', mse) 7. 使用模型进行预测 new_data = np.array([[1, 2, 3], [4, 5, 6]]) # 请替换为你要预测的新数据 predictions = model.predict(new_data) print('Predictions:', predictions)
你需要将上述代码中的your_data.csv、feature1、feature2、feature3和target替换为你实际的数据文件路径和列名。
本站发布或转载的文章及图片均来自网络,其原创性以及文中表达的观点和判断不代表本站,有问题联系侵删!
本文链接:http://www.xixizhuji.com/fuzhu/38914.html