From e4e667c572e28a01545a986f8038e2538ebf400c Mon Sep 17 00:00:00 2001
From: Caroline DE POURTALES <cdepourt@montana.irit.fr>
Date: Tue, 12 Jul 2022 16:31:38 +0200
Subject: [PATCH] assets
---
callbacks.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/callbacks.py b/callbacks.py
index 8213b61..18d5e49 100644
--- a/callbacks.py
+++ b/callbacks.py
@@ -12,6 +12,7 @@ from utils import extract_data
from pages.application.RandomForest.utils import xrf
from pages.application.RandomForest.utils.xrf import *
+
sys.modules['xrf'] = xrf
from sklearn.ensemble._voting import VotingClassifier
@@ -255,7 +256,7 @@ def register_callbacks(app):
try:
if ml_type is None:
return warning_selection_model, None, {}, {}
- elif pretrained_model is None :
+ elif pretrained_model is None:
return warning_selection_pretrained_model, None, {}, {}
else:
pretrained_model = parse_contents_graph(pretrained_model, model_filename)
@@ -337,11 +338,11 @@ def register_callbacks(app):
if ml_type == "RandomForest":
pretrained_model = parse_contents_graph(pretrained_model, model_filename)
if isinstance(pretrained_model, xrf.rndmforest.RF2001):
- return int(pretrained_model.forest.n_estimators)
+ return int(pretrained_model.forest.n_estimators) - 1
elif isinstance(pretrained_model, RandomForestClassifier):
- return pretrained_model.n_estimators
+ return pretrained_model.n_estimators - 1
elif isinstance(pretrained_model, VotingClassifier):
- return len(pretrained_model.estimators)
+ return len(pretrained_model.estimators) - 1
else:
return 0
--
GitLab