AutoML

对于AutoML自动机器学习,传统框架有auto-sklearntpoth2oauto_ml等;深度神经网络框架有Auto-Keras等。

根据评测

auto-sklearn在分类问题表现最好,tpot在回归问题最佳。

另一个评测也显示auto-sklearn和tpot性能不错。


auto-sklearn不支持windows,且在集群上离线安装困难;tpot支持windows,集群离线安装简单,还可以conda方式安装。推荐tpot

linux下安装及测试,

# install
tarj tpot-0.11.5.tar.gz
cd tpot-0.11.5/
pip install deap-1.3.1-cp38-cp38-manylinux2010_x86_64.whl
pip install update_checker-0.18.0-py3-none-any.whl
pip install stopit-1.1.2.tar.gz
pip install -e .

# test
from tpot import TPOTClassifier

windows下安装,

conda install -c conda-forge tpot