DIA
/
enose_2025
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

main
ifiguero 2025-03-20 13:00:14 -03:00
parent f7bee825a3
commit 9b5211567a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TrainerClass.py
View File

@ -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):