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

一、引言

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

小明看上一套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万入手显然是不太适合投资的!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,937评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,503评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,712评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,668评论 1 276
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,677评论 5 366
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,601评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,975评论 3 396
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,637评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,881评论 1 298
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,621评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,710评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,387评论 4 319
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,971评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,947评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,189评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 44,805评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,449评论 2 342

推荐阅读更多精彩内容