auto-sklearn简介

来自官网首页

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]上进行联系

同时建议你在开发新的功能时,请先创建新的发展分支,同时在所有的测试结束并通过后,进行项目合并.