AutoX是什么?

July 8, 2022 · View on GitHub

English | 简体中文 logo

AutoX是什么?

AutoX一个高效的自动化机器学习工具。 它的特点包括:

  • 效果出色: AutoX在多个kaggle数据集上,效果显著优于其他解决方案(见效果对比)。
  • 简单易用: AutoX的接口和sklearn类似,方便上手使用。
  • 通用: 适用于分类和回归问题。
  • 自动化: 无需人工干预,全自动的数据清洗、特征工程、模型调参等步骤。
  • 灵活性: 各组件解耦合,能单独使用,对于自动机器学习效果不满意的地方,可以结合专家知识,AutoX提供灵活的接口。
  • 比赛上分点总结:整理并公开历史比赛的上分点。

AutoX包含什么内容

加入社区

AutoX社区

框架

autox_competition

autox_competition framework

autox_recommend

autox_recommend framework

autox_video

autox_video framework

如何为AutoX贡献

如何为AutoX贡献

目录

安装

github仓库安装

git clone https://github.com/4paradigm/autox.git
pip install ./autox

pip安装

## pip安装包可能更新不及时,建议用github安装方式安装最新版本
!pip install automl-x -i https://www.pypi.org/simple/

快速上手

社区案例

汽车销量预测

比赛案例

见demo文件夹

数据集下载链接:https://pan.baidu.com/s/1p38OuP8_FJp2P_wJwhdFiw?pwd=8mxf

效果对比

不同任务下的效果提升百分比

data_type对比AutoGluon对比H2o
binary classification20.44%2.98%
regression37.54%39.66%
time-series28.40%32.46%

详细数据集对比

data_typesingle-or-multidata_namemetricAutoXAutoGluonH2o
binary classificationsingle-tableSpringleafauc0.788650.611410.78186
binary classification-nlpsingle-tablestumbleuponauc0.871770.810250.79039
binary classificationsingle-tablesantanderauc0.891960.646430.88775
binary classificationmulti-tableIEEEaccuracy0.9208090.7249250.907818
regressionsingle-tableventilatormae0.7558.4344.221
regressionsingle-tableAllstate Claims Severitymae1137.078851173.359171163.12014
regressionsingle-tablezhidemaimse1.00341.94661.1927
regressionsingle-tableTabular Playground Series - Aug 2021rmse7.8773110.39447.8895
regressionsingle-tableHouse Pricesrmse0.130430.131040.13161
regressionsingle-tableRestaurant Revenuermse2133204.3214631913829.5987628958013.69639
regressionmulti-tableElo Merchant Category Recommendationrmse3.722283.8080122.88899
regression-tssingle-tableDemand Forecastingsmape13.7924125.3918218.89678
regression-tsmulti-tableWalmart Recruitingwmae4660.991745024.161795128.31622
regression-tsmulti-tableRossmann Store SalesRMSPE0.138500.204530.35757
regression-cvsingle-tablePetFinderrmse20.132723.173221.0586

AutoX成就

企业支持

比赛获奖

TODO

功能开发完成后,发布相应的使用demo

  • 多分类任务

若有其他希望AutoX支持的功能,欢迎提issue! 欢迎填写用户调研问卷,让AutoX变得更好!

错误排查

错误信息解决办法