Would be nice to add tests, both for the classifier and regressor variants.
Would be nice to add tests, both for the classifier and regressor variants.