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

该博客介绍了如何使用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):
    """
    利用训练数据,估计模型参数
    
    参数
    -
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值