From 9b5211567a7b152f08629dbfdb1ec6bf441a2f46 Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Thu, 20 Mar 2025 13:00:14 -0300 Subject: [PATCH] fix --- TrainerClass.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/TrainerClass.py b/TrainerClass.py index e8d0389..da2b153 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -540,7 +540,7 @@ class eNoseTrainer: if Y_discrete.ndim == 2: Y_discrete = np.sum(Y_discrete, axis=1) - dataset = f'Conv1d-w{window}-{self.loader.smooth}' + dataset = f'Conv1D-w{window}-{self.loader.smooth}' for i, (train_index, test_index) in enumerate(gss.split(X_conv1d, Y_discrete, G_conv1d)): self.logger.info("{:=^60}".format(f'CV {i+1}/{int(1/self.ratio)} {dataset}')) @@ -554,7 +554,7 @@ class eNoseTrainer: self.logger.debug(f"Y_test: {Y_test.shape}") - model_id = "Conv1d_v1_{}".format(i) + model_id = "Conv1D_v1_{}".format(i) self.trained += 1 if self.row_exists(dataset, model_id):