patch

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']

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