Implementation

Python implementation with Criterion implemented in Cython!

scikit-learn compatible

We rolled our own scikit-learn compatible estimator following the Rolling your own estimator instructions and using the provided project template from scikit-learn.

We are trying to be consistent with scikit-learn’s decision tree and ensemble modules.

Exceptions

Used class_balance as hyperparameter name instead of class_weight

The class_weight hyperparameter name is recognized by check_estimator() and the test check_class_weight_classifiers() is performed that uses the dict parameter and requires for a decision tree the “min_weight_fraction_leaf” hyperparameter to be implemented to pass the test.

Basic Concepts

The basic concepts for the implementation of the classifiers are based on:

7. Louppe, Understanding Random Forests, PhD Thesis, 2014