crflynn / skranger

scikit-learn compatible Python bindings for ranger C++ random forest library
https://skranger.readthedocs.io/en/stable/
GNU General Public License v3.0
52 stars 7 forks source link

add check_estimator tests #72

Closed crflynn closed 3 years ago

crflynn commented 3 years ago

Add check_estimator tests to classifier and regressor. We can't test the survival forest because it conforms to the sksurv api rather than sklearn's.

In doing so we add validation to ensure check_estimator passes. We also add logic to ensure that passing sample_weight of ones is equivalent to passing None.

Changes:

https://scikit-learn.org/stable/modules/generated/sklearn.utils.estimator_checks.check_estimator.html