Upgrade to Pro — share decks privately, control downloads, hide ads and more …

Engineering Scikit-learn (NIPS 2014 Workshop)

Engineering Scikit-learn (NIPS 2014 Workshop)

The software development and organizational principles behind the scikit-learn project.

Andreas Mueller

August 16, 2015
Tweet

More Decks by Andreas Mueller

Other Decks in Programming

Transcript

  1. 5 Goal: High quality, easy to use machine learning library.

    Keep it usable, keep it maintainable.
  2. 8 Non-Goals Non-programmatic interfaces Algorithm development Cutting edge algorithms Structured,

    online, or reinforcement learning. “I thought it was more like CRAN”
  3. 10

  4. 11 We’ve been using it quite a lot for music

    recommendations at Spotify and I think it’s the most well-designed ML package I’ve seen so far. - spotify scikit-learn in one word: Awesome. - machinalis I’m constantly recommending that more developers and scientists try scikit-learn. - lovely The documentation is really thorough, as well, which makes the library quite easy to use. - OkCupid scikit-learn makes doing advanced analysis in Python accessible to anyone. - yhat
  5. 16 Sensible Defaults Everything is default constructible! for clf in

    [KneighborsClassifier(), SVC(), DecisionTreeClassifier(), RandomForestClassifier(), AdaBoostClassifier(), GaussianNB(), LDA(), QDA()]: clf.fit(X_train, y_train) print(clf.score(X_test, y_test))
  6. 17 Common Tests classifiers = all_estimators(type_filter='classifier') for name, Classifier in

    classifiers: # test classfiers can handle non-array data yield check_classifier_data_not_an_array, name, Classifier # test classifiers trained on a single label # always return this label yield check_classifiers_one_label, name, Classifier yield check_classifiers_classes, name, Classifier yield check_classifiers_pickle, name, Classifier yield check_estimators_partial_fit_n_features, name, Classifier
  7. 18 Flat Class Hierarchy, Few Types • Numpy arrays /

    sparse matrices • Estimators • [Cross-validation objects] • [Scorers]
  8. 24 Multi-Platform Support • Linux / Mac / Windows /

    Solaris (no kidding) • 32bit / 64bit • Python2.6 / Python2.7 / Python 3.4 • GCC, Clang, MSVC • Blas dependency... • And we want “one click” install
  9. 27 Backward compatibility from sklearn.cross_validation import Bootstrap Bootstrap(10) sklearn/cross_validation.py:685: DeprecationWarning:

    Bootstrap will no longer be supported as a cross-validation method as of version 0.15 and will be removed in 0.17.