Auto-PyTorch

Auto-PyTorch 是一个基于 PyTorch 的自动化机器学习工具包

>>> import autoPyTorch
>>> cls = autoPyTorch.api.tabular_classification.TabularClassificationTask()
>>> cls.search(X_train, y_train)
>>> predictions = cls.predict(X_test)

Auto-PyTorch 将机器学习用户从算法选择和超参数调优中解放出来。它利用了最近在贝叶斯优化元学习集成构建方面的进展。通过阅读我们的论文Auto-PyTorch Tabular: 用于高效和稳健的 AutoDL 的多保真元学习,了解更多关于Auto-PyTorch的信息。

示例

许可

Auto-PyTorch 的许可方式与 scikit-learn 相同,即 3 条款的 BSD 许可。

引用 Auto-PyTorch

如果您在科学出版物中使用 Auto-PyTorch,我们希望您引用以下论文

Auto-PyTorch 表格:用于高效和稳健的 AutoDL 的多保真元学习,

Bibtex 条目

@article{zimmer2020auto,
   title={Auto-pytorch tabular: Multi-fidelity metalearning for efficient and robust autodl},
   author={Zimmer, Lucas and Lindauer, Marius and Hutter, Frank},
   journal={arXiv preprint arXiv:2006.13799},
   year={2020}
}

贡献

我们感谢对 Auto-PyTorch 的所有贡献,包括错误报告、文档和新功能。如果您想贡献代码,可以从标记为 需要贡献者问题跟踪器 中选择一个问题。

注意

为了避免在重复工作或不太可能合并的功能上花费时间,强烈建议您在开始工作之前通过开启一个 github issue 联系开发者。

在开发新功能时,请从 refactor_development 分支创建一个新分支。提交拉取请求时,请确保所有测试都通过。