千锋教育-做有情怀、有良心、有品质的职业教育机构

手机站
千锋教育

千锋学习站 | 随时随地免费学

千锋教育

扫一扫进入千锋手机站

领取全套视频
千锋教育

关注千锋学习站小程序
随时随地免费学习课程

当前位置:首页  >  技术干货  > python机器学习线性回归算法

python机器学习线性回归算法

来源:千锋教育
发布人:zyh
时间: 2023-05-31 15:00:00 1685516400

  在 Python 中实现线性回归算法有多种方式,以下是一种常用的方法:

  1. 数据准备:将数据集划分为训练集和测试集,并进行特征缩放(将数据标准化)。

  2. 导入必要的库: NumPy 和 Pandas 用于数据操作, Matplotlib 用于可视化。

  3. 引入线性回归模型:引入 Scikit-Learn 中的线性回归模型,如下所示:  

from sklearn.linear_model import LinearRegression

  4. 创建线性回归对象:创建 LinearRegression() 对象。  

lin_reg = LinearRegression()

  5. 训练模型:使用训练集进行训练。 

lin_reg.fit(X_train, y_train)

  其中,X_train 是训练集特征,y_train 是训练集标签。

  6. 预测结果:使用测试集进行预测。 

y_pred = lin_reg.predict(X_test)

  其中,y_pred 是测试集的预测标签。

  7. 可视化:使用 Matplotlib 可视化结果。 

plt.scatter(X_test, y_test, color='red')
plt.plot(X_test, y_pred, color='blue')
plt.title('Linear Regression')
plt.xlabel('Independent Variable')
plt.ylabel('Dependent Variable')
plt.show()

  完整代码示例: 

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression

# 读取数据
dataset = pd.read_csv('data.csv')

# 准备数据
X = dataset.iloc[:, :-1].values
y = dataset.iloc[:, 1].values
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=1/3, random_state=None)

# 特征缩放
from sklearn.preprocessing import StandardScaler
sc_X = StandardScaler()
X_train = sc_X.fit_transform(X_train)
X_test = sc_X.transform(X_test)

# 创建线性回归对象
lin_reg = LinearRegression()

# 训练模型
lin_reg.fit(X_train, y_train)

# 预测结果
y_pred = lin_reg.predict(X_test)

# 可视化
plt.scatter(X_test, y_test, color='red')
plt.plot(X_test, y_pred, color='blue')
plt.title('Linear Regression')
plt.xlabel('Independent Variable')
plt.ylabel('Dependent Variable')
plt.show()

  注:data.csv 数据集应该包含两列数据,第一列为特征,第二列为标签。

tags:
声明:本站稿件版权均属千锋教育所有,未经许可不得擅自转载。
10年以上业内强师集结,手把手带你蜕变精英
请您保持通讯畅通,专属学习老师24小时内将与您1V1沟通
免费领取
今日已有369人领取成功
刘同学 138****2860 刚刚成功领取
王同学 131****2015 刚刚成功领取
张同学 133****4652 刚刚成功领取
李同学 135****8607 刚刚成功领取
杨同学 132****5667 刚刚成功领取
岳同学 134****6652 刚刚成功领取
梁同学 157****2950 刚刚成功领取
刘同学 189****1015 刚刚成功领取
张同学 155****4678 刚刚成功领取
邹同学 139****2907 刚刚成功领取
董同学 138****2867 刚刚成功领取
周同学 136****3602 刚刚成功领取
相关推荐HOT