Benjamin Trent fa30246937

[ML] fixes decision tree classifier upload to account for probabilities (#465) ...

This switches our sklearn.DecisionTreeClassifier serialization logic to account for multi-valued leaves in the tree.

The key difference between our inference and DecisionTreeClassifier, is that we run a softMax over the leaf where sklearn simply normalizes the results.

This means that our "probabilities" returned will be different than sklearn.

2022-05-17 08:11:20 -04:00

..

pytorch

[ML] improve general pytorch model import and add tests (#463)

2022-05-05 10:50:53 -04:00

test_ml_model_pytest.py

[ML] fixes decision tree classifier upload to account for probabilities (#465)

2022-05-17 08:11:20 -04:00

test_optional_dependency_pytest.py

Move tests directory outside of eland namespace

2020-11-16 11:30:41 -06:00