From ca5d00805fbc089a5ad1980c64a3cb77bcd76029 Mon Sep 17 00:00:00 2001 From: Charly Lamothe <charly.lamothe@univ-amu.fr> Date: Fri, 6 Mar 2020 06:32:03 +0100 Subject: [PATCH] Add missing import in trainer --- code/bolsonaro/trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py index 34b76c8..6ec3909 100644 --- a/code/bolsonaro/trainer.py +++ b/code/bolsonaro/trainer.py @@ -2,6 +2,7 @@ from bolsonaro.models.model_raw_results import ModelRawResults from bolsonaro.models.omp_forest_regressor import OmpForestRegressor from bolsonaro.models.omp_forest_classifier import OmpForestBinaryClassifier, OmpForestMulticlassClassifier from bolsonaro.models.similarity_forest_regressor import SimilarityForestRegressor +from bolsonaro.models.kmeans_forest_regressor import KMeansForestRegressor from bolsonaro.error_handling.logger_factory import LoggerFactory from bolsonaro.data.task import Task from . import LOG_PATH -- GitLab