From fe6f7ffd6d4ef4347890aaeaad09483f7e3cc492 Mon Sep 17 00:00:00 2001 From: martin-sicho Date: Thu, 8 Feb 2024 14:51:11 +0100 Subject: [PATCH] use correct summary method --- qsprpred/models/tests.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/qsprpred/models/tests.py b/qsprpred/models/tests.py index 4c8de6df..245aea63 100644 --- a/qsprpred/models/tests.py +++ b/qsprpred/models/tests.py @@ -23,6 +23,7 @@ from sklearn.svm import SVC, SVR from xgboost import XGBClassifier, XGBRegressor +from .assessment.classification import create_metrics_summary from .assessment.regression import create_correlation_summary from ..models.early_stopping import EarlyStopping, EarlyStoppingMode, early_stopping from ..models.metrics import SklearnMetrics @@ -371,7 +372,7 @@ def testRandomForestClassifierFitWithSeed(self): parameters=parameters, ) self.fitTest(model, dataset) - expected_summary = self.createMetricsSummary(model) + expected_summary = create_metrics_summary(model) # Generate summary again, check that the result is identical model = self.getModel( @@ -380,7 +381,7 @@ def testRandomForestClassifierFitWithSeed(self): parameters=parameters, ) self.fitTest(model, dataset) - summary = self.createMetricsSummary(model) + summary = create_metrics_summary(model) self.assertListEqual(summary["Metric"], expected_summary["Metric"]) self.assertListEqual(summary["Model"], expected_summary["Model"])