gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-18 05:18:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(Project): m1 support (#233)

Browse Source
This commit is contained in:
Mark Aron Szulyovszky
2022-03-10 13:07:41 +01:00
committed by GitHub
parent 567cd5e9f0
commit 4a9f74d645
5 changed files with 23 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/model_map.py
+2 -3
View File
@@ -1,6 +1,5 @@
from models.sklearn import SKLearnModel
from sklearnex.ensemble import RandomForestClassifier
from sklearnex.ensemble import RandomForestRegressor
from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
from .base import Model
default_feature_selector_classification = SKLearnModel(
@@ -24,7 +23,7 @@ def get_model(model_name: str) -> Model:
)
)
elif model_name == "LogisticRegression_three_class":
from sklearnex.linear_model import LogisticRegression as LogisticRegression_EX
from sklearn.linear_model import LogisticRegression as LogisticRegression_EX
return set_name(
SKLearnModel(