auto_ml

收录理由

auto_ml 是一个基于 scikit-learn 构建的早期 AutoML 库，虽然开发在 2021 年已经停止，但它的代码仍然值得一读。使用时只需提供一个 pandas DataFrame，并简单说明哪些列是类别型、哪一列是预测目标，它就会自动在多种回归器和分类器之间搜索，拟合出一个可用的模型，省去大量手工调参的麻烦。它真正特别的地方在于对生产环境的考虑：训练好的模型可以序列化保存到磁盘，对单条字典数据的预测大约只需一毫秒；另外，它的类别集成特性允许你为每个门店或客户单独训练一个模型，却通过一个统一的 API 来调用，省去了管理成千上万个模型文件的负担。对于想了解现代框架出现之前 AutoML 管道是如何工作的数据科学家，或者想找一个把 XGBoost、LightGBM、CatBoost 和 Keras 整合进同一训练循环的简洁示例的人来说，这份代码能带来实实在在的启发。可以把它当作学习资料和设计参考，而不是用来搭建当前生产服务的基石。