search

intel / scikit-learn-intelex

Intel(R) Extension for Scikit-learn is a seamless way to speed up your Scikit-learn application
https://intel.github.io/scikit-learn-intelex/
Apache License 2.0
1.23k stars 175 forks source link

Sklearn integration of RF predict uses float fptype always for daal4py #206

Closed SmirnovEgorRu closed 3 years ago

SmirnovEgorRu commented 4 years ago

https://github.com/IntelPython/daal4py/blob/ca67cc2d070ff27318e210e4b74f05ab7b9fa1d0/daal4py/sklearn/ensemble/decision_forest.py#L285

Looking at the code - we use float type always for RF predict. Probably, it's handled in another place, but it looks like an issue.

PivovarA commented 3 years ago

Done