From 7f1f29d78f129f88bec089eb28d78fdbc92a7713 Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Mon, 10 Mar 2025 19:02:50 -0300 Subject: [PATCH] patch --- LoaderClass.py | 3 +++ TrainerClass.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/LoaderClass.py b/LoaderClass.py index 83610ec..bd8376b 100644 --- a/LoaderClass.py +++ b/LoaderClass.py @@ -314,6 +314,9 @@ class GasSensorDataLoader: x_output = np.concatenate((x_output, x_sample)) y_output = np.concatenate((y_output, y_sample)) + target_scaler = MinMaxScaler() + y_output = target_scaler.fit_transform(y_output) + self.dataset['xboost'] = (x_output, y_output, g_output) return self.dataset['xboost'] diff --git a/TrainerClass.py b/TrainerClass.py index 9be995c..f6027e8 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -5,7 +5,7 @@ import matplotlib.pyplot as plt import matplotlib matplotlib.rcParams['text.usetex'] = True -from sklearn.model_selection import GroupShuffleSplit, ShuffleSplit, GridSearchCV +from sklearn.model_selection import StratifiedGroupKFold, StratifiedShuffleSplit, GridSearchCV from sklearn.metrics import mean_squared_error, mean_absolute_error from xgboost import XGBRegressor