来自官网首页
auto-sklearn是什么?
auto-sklearn是一个自动化机器学习的工具包,其基于sklearn编写.
>>> import autosklearn.classification
>>> cls = autosklearn.classification.AutoSklearnClassifier()
>>> cls.fit(X_train, y_train)
>>> predictions = cls.predict(X_test)
auto-sklearn可以进行机器学习算法的自动选择与超参数的自动优化,它使用的技术包括贝叶斯优化,元学习,以及集成机构?(ensemble construction).你可以通过这篇文章,NIPS 2015来学习关于更多auto-sklearn背后的原理与技术.
例子
>>> import autosklearn.classification
>>> import sklearn.model_selection
>>> import sklearn.datasets
>>> import sklearn.metrics
>>> X, y = sklearn.datasets.load_digits(return_X_y=True)
>>> X_train, X_test, y_train, y_test = \
sklearn.model_selection.train_test_split(X, y, random_state=1)
>>> automl = autosklearn.classification.AutoSklearnClassifier()
>>> automl.fit(X_train, y_train)
>>> y_hat = automl.predict(X_test)
>>> print("Accuracy score", sklearn.metrics.accuracy_score(y_test, y_hat))
如果将上面的代码运行一个小时,那么其精度将会高于0.98.
手册
- 安装
- 手册
- api
- 扩展auto-sklearn
手册中文翻译
许可证
auto-sklearn与scikit-sklearn的许可证一样,即都为三条款的BSD许可
援引auto-sklearn
如果你在科学出版物上使用auto-sklearn,我们将感激不尽
Efficient and Robust Automated Machine Learning, Feurer et al., Advances in Neural Information Processing Systems 28 (NIPS 2015).
Bibtex entry:
@incollection{NIPS2015_5872,
title = {Efficient and Robust Automated Machine Learning},
author = {Feurer, Matthias and Klein, Aaron and Eggensperger, Katharina and
Springenberg, Jost and Blum, Manuel and Hutter, Frank},
booktitle = {Advances in Neural Information Processing Systems 28},
editor = {C. Cortes and N. D. Lawrence and D. D. Lee and M. Sugiyama and R. Garnett},
pages = {2962--2970},
year = {2015},
publisher = {Curran Associates, Inc.},
url = {http://papers.nips.cc/paper/5872-efficient-and-robust-automated-machine-learning.pdf}
}
贡献
我们感谢所有对auto-sklearn做出贡献的人,无论你是写的bug报告还是文档,亦或是新的贡献.同时如果你想要贡献代码.你可以使用issue tracker
同时为了项目合并前避免重复的工作,强烈建议你在进行工作前与我们的工作人员在(github issues)[https://github.com/automl/auto-sklearn/issues]上进行联系
同时建议你在开发新的功能时,请先创建新的发展分支,同时在所有的测试结束并通过后,进行项目合并.