From 7607006b95fcd555a6a8e846eb4cd33f4702914a Mon Sep 17 00:00:00 2001 From: Oege Dijk Date: Thu, 27 May 2021 11:11:59 +0200 Subject: [PATCH] add sklearn imports to businessrule module --- rule_estimator/businessrule.py | 3 +++ rule_estimator/estimators.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/rule_estimator/businessrule.py b/rule_estimator/businessrule.py index 350ffce..e8afa34 100644 --- 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 index 82077b0..dc54167 100644 --- 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