美文网首页
iOS 机器学习(Core ML)--(二)建立单因子线性回归模

iOS 机器学习(Core ML)--(二)建立单因子线性回归模

作者: 枫叶情结 | 来源:发表于2023-03-10 14:29 被阅读0次

一、引言

故事从小明要买房开始说起!

小明看上一套100平的房子,房子售价是120万元,小明不知道这套房子120万是贵了还是便宜了,为了搞清楚这套房子是否适合投资,小明收集了一些这套房子所在区域的房价(忽略其他因素,仅考虑面积),收集到的数据如下:


房价.png

接下来让我们利用机器学习,帮小明评估下这套房子120万是否适合投资吧!

二、环境准备

安装 Anaconda、Jupyter Notebook、coremltools、Python、pandas、numpy、Matplotlib

Anaconda:python包管理和环境管理软件

Jupyter Notebook:Web应用程序,可以实时写代码、运行代码、查看结果,并可视化数据

coremltools:苹果提供的Python工具包,可用于模型转换

pandas:用于处理数据的python包

numpy:用于矩阵运算的python包

Matplotlib:用于画图的python包

三、建立单因子线性回归模型

1、数据可视化

先把小明采集到的数据画出来,如下图所示:


数据可视化.png

2、选择合适算法

由上图可知,房价与面积大致是呈线性正相关的,即面积越大,售价越高。我们决定采用线性回归模型来解决这个问题。假设面积为x,售价为y,则 x 和 y 之间应该是存在一个定量关系( y = ax + b)。因此,我们只要找出a 和 b的值,就能根据面积来预测房价。

3、模型好坏评估

关系式 y = ax + b 中,a 和 b有无数种可能,到底a 和 b的值取多少才能最大程度拟合上面的点呢?也就是说a 和 b的值是多少才能使得预测出来的y'值与实际的y尽可能的接近?我们可以用损失函数均方误差 MSER2 分数等来评估,公式如下:

损失函数展开.png MES.png R2分数.png

由上面的公式可知,损失函数或MSE越小越好,R2分数越接近1越好。

4、寻找最小损失函数

我们知道了如何评估模型的好坏后,就可以对不同的a 和 b的值进行评估了。可以通过梯度下降法来找出最小损失函数!


梯度下降法.png

梯度下降法就是对损失函数(不同a 和 b的值)不断求导,直至收敛,寻找出损失函数最小值的一个过程。当损失函数导数不再变化了,也就找到了最合理的 a 和 b 的值了。

简单的理解,就好比你在山顶上,要找到一条下山最快的路,那么你可以走一步,然后四处张望,找到一个最陡峭的方向(导数最大),然后向前走一步,并不断的重复上述动作。

5、核心代码

#建立线性回归模型
from sklearn.linear_model import LinearRegression
model = LinearRegression() #创建一个实例

#模型训练
model.fit(x,y) #模型训练

#获取线性回归模型的参数a b
a = model.coef_
b = model.intercept_

#预测 (x=100,计算y)
x_test = np.array([[100]])
y_test_predict = model.predict(x_test)

#模型评估
from sklearn.metrics import mean_squared_error,r2_score
MSE = mean_squared_error(y,y_predict) #均方差(相当于损失函数)
R2 = r2_score(y,y_predict) #R2分数 = 1 - MSE/方差   R2越接近1,说明模型越优秀

本例计算出来的R2分数是0.806

模型预测线如下:


预测线.png

可见预测出来的值还是比较合理的。

6、导出iOS可使用的模型

使用coremltools将模型导成 Core ML 模型格式(.mlmodel),代码如下:

#设置模型属性
input_features = "acreage"
output_feature = "price"

#转成iOS可用的模型
model = coremltools.converters.sklearn.convert(model, input_features, output_feature)
model.save("HJQHousePrice.mlmodel")

最终的模型如下:


模型.png

7、使用模型

如何使用见上一章

预测结果如下:

预测结果.png

由预测结果可知,小明看上的这套100平的房子预估合理价格是94万,以120万入手显然是不太适合投资的!

相关文章

  • 训练你的第一个机器学习模型

    在之前的文章手把手教你在应用里用上iOS机器学习框架Core ML里,我们用苹果的机器学习框架Core ML和事先...

  • 线性回归模型

    参考:1.使用Python进行线性回归2.python机器学习:多元线性回归3.线性回归概念 线性回归模型是线性模...

  • 常见回归算法

    sklearn作为机器学习中一个强大的算法包,内置了许多经典的回归算法。 线性回归 线性回归拟合一个带系数的线性模...

  • 2019-10-29

    Day2 简单线性回归模型 机器学习入门--简单线性回归机器学习算法之线性回归算法 导入库matplotlib 绘...

  • 手把手教你在应用里用上iOS机器学习框架Core ML

    2017年的WWDC上,苹果发布了Core ML这个机器学习框架。现在,开发者可以轻松的使用Core ML把机器学...

  • Python编写线性回归算法

    Python编写线性回归算法 前言 线性回归(Linear Regression)是机器学习的基础,作为机器学习算...

  • 如何快速理解线性回归及模型训练?

    在机器学习领域内,经常会听到回归问题、线性回归等专业术语,那什么是回归问题,线性回归又是什么意思以及怎么训练一个模...

  • iOS Core ML框架机器学习

    iOS 11.0+ //目前iOS低版本系统不支持 macOS 10.13+ 参考苹果官方文档Core ML ...

  • iOS Core ML 机器学习入门

    前言机器学习的分类: 有监督学习准确 但是 比较费时间给机器一个对应关系(训练数据)比如告诉它手机是什么(给训练的...

  • 使用Core ML构建更具智能的应用

    前几天WWDC上,苹果发布了机器学习框架Core ML。根据文档,Core ML是基于Metal和Accelera...

网友评论

      本文标题:iOS 机器学习(Core ML)--(二)建立单因子线性回归模

      本文链接:https://www.haomeiwen.com/subject/kuctrdtx.html