search

microsoft / hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.
MIT License
3.32k stars 274 forks source link

No module named 'sklearn._loss.link' #678

Closed Himanshuengg closed 1 year ago

Himanshuengg commented 1 year ago

ModuleNotFoundError Traceback (most recent call last) in 1 import numpy as np 2 from sklearn.ensemble import RandomForestClassifier ----> 3 from hummingbird.ml import convert 4 5 # Create some random data for binary classification

3 frames /usr/local/lib/python3.8/dist-packages/hummingbird/ml/operator_converters/sklearn/linear.py in 11 import numpy as np 12 from onnxconverter_common.registration import register_converter ---> 13 from sklearn._loss.link import LogLink 14 15 from .._linear_implementations import LinearModel

ModuleNotFoundError: No module named 'sklearn._loss.link'

NOTE: If your import is failing due to a missing package, you can manually install dependencies using either !pip or !apt.

mshr-h commented 1 year ago

Which sklearn version do you use? You need at least 1.1.0 or above.

Himanshuengg commented 1 year ago

Thanks for the information. After updating sklearn version my issue has been resolved.