搜文章
推荐 原创 视频 Java开发 iOS开发 前端开发 JavaScript开发 Android开发 PHP开发 数据库 开发工具 Python开发 Kotlin开发 Ruby开发 .NET开发 服务器运维 开放平台 架构师 大数据 云计算 人工智能 开发语言 其它开发
Lambda在线 > 量子位 > 调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

量子位 2020-08-02
萧箫 发自 凹非寺
量子位 报道 | 公众号 QbitAI

Python开源机器学习建模库PyCaret,刚刚发布了2.0版本。

这款堪称「调包侠神器」的模型训练工具包,几行代码就能搞定模型编写、改进和微调。

从数据预处理到模型效果对比,PyCaret都能自动实现。

所以,PyCaret长啥样,2.0的版本又做了什么改进?

一起来看看。

机器学习库的「炼丹炉」

PyCaret说白了,有点像一个机器学习库的炼丹炉。

以下是它「熔」进来的部分库:

数据处理:pandas、numpy…
数据可视化:matplotlib、seaborn…
各种模型:sklearn、xgboost、catboost、lightgbm…

嗯…sklearn直接就给封装进去了,调用很方便。

然后,PyCaret这个炼丹炉,自带功能“按键”(定义了一些函数),包括数据预处理、模型训练、模型集成、模型分析、模型测试等。

只需要写上几行Python代码,这些功能“按键”就会被按下,PyCaret自动帮你实现。

至于实现过程中需要调用什么基本库,那些可以放弃思考不需要考虑。

从下图来看,仅仅是预处理阶段,就包含样本划分、数据预处理、缺失值处理、归一化、独热编码等功能。

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

如果要实现必需的预处理功能,需要多少行代码来调用?

答案是0行。

因为,当使用setup()进行初始化时,PyCaret将自动执行机器学习必需的数据预处理步骤,包括缺失值插入、分类变量编码、标签编码、数据集拆分等。

例如,在数据处理前,你发现数据集有空缺的地方(下图中NaN部分)

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

别怕,PyCaret会自动分析数据,进行缺失值插入。

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

预处理后,PyCaret还贴心地帮你准备了各种模型。

从模型训练、选用到测试,只有你想不到的,没有它做不了的。

如果已经将数据处理好,并交给PyCaret,一个compare_models函数就能训练库中的所有模型,进行结果比较后,标出最佳模型。

如下图,各种模型指标的最优值会被一键标黄,就看你怎么选择了。

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

选好后,想对模型进行一点优化?一个tune_model函数就能帮你搞定。

或者,不想仅仅选用一个模型?

PyCaret也准备了模型集成的函数,blend和stack任你选。

除此之外,模型参数的分析(包括可视化)也只需要几行代码就能实现,功能非常强大。

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

最后,PyCaret还能为新数据提供迭代预测结果,下面的效果,同样只需要几行代码就能完成。

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

那么,这次PyCaret增强,进行了什么改进呢?(项目见传送门)

PyCaret 2.0增强版

这是PyCaret 2.0的6大特色,有些在1.0就有了,有些功能如实验日志,看起来是更新后新加入的功能。

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

实验日志,对于模型的调整不可或缺。

例如,想要将训练过程中模型的精度变化可视化,通常我们会在模型中加入生成日志文件的函数,生成一个更直观的时间-精度变化图。

PyCaret 2.0加入了实验日志的功能,自动帮你跟踪模型实验过程中的各项指标,以及生成视觉效果等。

不仅如此,在2.0中,模型生成到预测的所有工作流程,现在可以被设计了。

也就是说,你可以设置一条自定义「流水线」,在这个过程中,从训练到测试,所有模块的功能都会被自动化完成。

甚至,PyCaret 2.0还提供了机器学习模型前端软件的搭建工具。

以及,PyCaret 2.0现在几乎支持所有算法的并行处理,xgboost和catboost模型也支持GPU训练。

调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码

除此之外,还有一些新的程序功能,等待你去发现。

传送门

参考链接:
https://towardsdatascience.com/announcing-pycaret-2-0-39c11014540e

完 —

本文系网易新闻•网易号特色内容激励计划签约账号【量子位】原创内容,未经账号授权,禁止随意转载。

报名 |「隐私计算+AI」技术直播

不了解任何隐私AI技术的情况下,开发者怎样做到只改动两三行代码,就将现有AI代码转换为具备数据隐私保护功能的程序?

8月6号, 中科院软件研究所博士、矩阵元算法科学家--谢翔将直播解析,加小助手“qbitbot9”或者直接扫码,即可进入直播交流群:

量子位 QbitAI · 头条号签约作者


վ'ᴗ' ի 追踪AI技术和产品新动态


喜欢就点「在看」吧 !


版权声明:本站内容全部来自于腾讯微信公众号,属第三方自助推荐收录。《调包侠神器2.0发布,Python机器学习模型搭建只需要几行代码》的版权归原作者「量子位」所有,文章言论观点不代表Lambda在线的观点, Lambda在线不承担任何法律责任。如需删除可联系QQ:516101458

文章来源: 阅读原文

相关阅读