线性回归模型机器学习代码实现

最新推荐文章于 2024-07-15 10:49:30 发布

weiyi99999

最新推荐文章于 2024-07-15 10:49:30 发布

阅读量1.4k

点赞数 3

分类专栏：数据科学文章标签： Python 线性回归机器学习

本文链接：https://blog.csdn.net/weiyi99999/article/details/80519039

版权

该博客介绍了如何使用Python进行线性回归模型的实现，包括数据划分、模型训练、模型评估和结果可视化。通过训练数据估计模型参数，并在测试数据集上计算均方差和决定系数来评价模型效果。

摘要由CSDN通过智能技术生成

模型实现需要四步：

1.将数据划分为训练集和测试数据集，解决过度拟合的问题

2.利用训练数据集训练模型，估计模型参数

3.利用测试数据集评价模型，计算对应的均方差和决定系数

4.用图像化的方式，展示模型效果

实现代码如下：

import os
import sys

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn import linear_model
#建立线性回归模型
def linearModel(data):
    """
    线性回归模型建模步骤展示
    
    参数
    ---
    data:DataFrame,建模数据
    """
    features = ["x"]
    labels = ["y"]
    #划分训练集和测试集
    trainData = data[:15]
    testData = data[15:]
    #产生并训练模型
    model = trainModel(trainData,features,labels)
    #评价模型效果
    error,score = evaluateModel(model,testData,features,labels)
    #图形化模型结果
    visualizeModel(model,data,features,labels,error,score)
    
#使用第三方开源算法库scikit-learn来搭建和训练线性回归模型
def trainModel(trainData,features,labels):
    """
    利用训练数据，估计模型参数
    
    参数
    -