对于AutoML自动机器学习,传统框架有auto-sklearn、tpot、h2o、auto_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