机器学习包sklearn的使用——以波士顿房价预测为例

一、 问题定义

本项目利用马萨诸塞州波士顿郊区的房屋信息数据训练和测试一个模型,并对模型的性能和预测能力进行测试。通过该数据训练后的好的模型可以被用来对房屋做特定预测—尤其是对房屋的价值。对于房地产经纪等人的日常工作来说,这样的预测模型被证明非常有价值。
本数据集的数据特征:
在这里插入图片描述

图1 波士顿房价数据集特征值 1
本项目的数据集来自UCI机器学习知识库。波士顿房屋这些数据于1978年开始统计,共506个数据点,涵盖了麻省波士顿不同郊区房屋14种特征的信息。本项目对原始数据集做了以下处理:
(1)在打乱后去 90%数据作为训练集,10%的数据作为测试集
(2)有16个’MEDV’ 值为50.0的数据点被移除。 这很可能是由于这些数据点包含遗失或看不到的值。
(3)有1个数据点的 ‘RM’ 值为8.78. 这是一个异常值,已经被移除。
(4)对于本项目,房屋的’RM’, ‘LSTAT’,'PTRATIO’以及’MEDV’特征是必要的,其余不相关特征已经被移除。
(5)'MEDV’特征的值已经过必要的数学转换,可以反映35年来市场的通货膨胀效应。
(6)由于这个项目最终目的是建立一个预测房价的模型,所以我将数据集分为特征(RM、LSTAT、PTRATIO)和目标变量(MEDV)
在这里插入图片描述
图2 需要关注的特征值

二、 实验原理与流程

2.1实验原理

第一步操作,就是在读入housing.csv文件后,统计一下价格的最小值、最大值、均值、中值和标准差,后期做数据分析的时候,其实用到这几个统计值。
这里实现呢,本身python是自带统计函数的,但是,这里使用numpy相关的函数处理,理论上速度应该更快一些。
可以这样认为,numpy包是基于python做的优化,而pandas是基于numpy做的优化。
第二步操作,开始建模准备
选用R^2, 也就是决定系数作为模型的评测函数,一看这意思,就是要用回归分析来进行预测啦,因为决定系数,就是回归分析中最经常用的评价指标,
比如࿰

  • 4
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
预测波士顿房价是一个经典的机器学习问题,也是许多人学习机器学习时的入门案例。 在Python中,我们可以使用scikit-learn(sklearn)这个机器学习库来进行波士顿房价的预测分析。下面我将简要介绍一下这个过程。 首先,我们需要导入相关的库和数据集: ```python import numpy as np import pandas as pd from sklearn.datasets import load_boston from sklearn.model_selection import train_test_split from sklearn.linear_model import LinearRegression from sklearn.metrics import mean_squared_error boston = load_boston() df = pd.DataFrame(boston.data, columns=boston.feature_names) df['target'] = boston.target ``` 接着,我们需要对数据进行预处理,括划分数据集、特征缩放等等: ```python X = df.drop('target', axis=1) y = df['target'] X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42) # 特征缩放 from sklearn.preprocessing import StandardScaler scaler = StandardScaler() X_train = scaler.fit_transform(X_train) X_test = scaler.transform(X_test) ``` 然后就可以开始建立模型并进行训练了。这里我们选择线性回归模型: ```python model = LinearRegression() model.fit(X_train, y_train) ``` 最后,我们可以使用测试集对模型进行评估: ```python y_pred = model.predict(X_test) mse = mean_squared_error(y_test, y_pred) print("MSE: ", mse) ``` 这样就可以得到模型的均方误差了。 当然,这只是一个简单的例子。在实际的应用中,我们可能需要使用更加复杂的模型,比如决策树、随机森林等等。同时,我们也需要在数据预处理、特征工程等方面进行更加深入的研究和实践。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值