add sklearn imports to businessrule module

oegedijk · May 27, 2021 · 7607006 · 7607006
1 parent d55647d
commit 7607006
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/rule_estimator/businessrule.py b/rule_estimator/businessrule.py
@@ -8,6 +8,9 @@
 import numpy as np
 import pandas as pd
 
+from sklearn.base import BaseEstimator
+from sklearn.metrics import accuracy_score, mean_squared_error
+
 from igraph import Graph
 
 from .storable import Storable

diff --git a/rule_estimator/estimators.py b/rule_estimator/estimators.py
@@ -11,7 +11,7 @@
 
 from sklearn.base import BaseEstimator
 from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
-from sklearn.metrics import accuracy_score, precision_score, recall_score, mean_squared_error
+from sklearn.metrics import accuracy_score, mean_squared_error
 
 from igraph import Graph