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

fix

main
ifiguero 2025-03-18 11:07:52 -03:00
parent c7ba91be06
commit 24375b392b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TrainerClass.py
View File

@ -535,11 +535,11 @@ class eNoseTrainer:
if Y_discrete.ndim == 2:
Y_discrete = np.sum(Y_discrete, axis=1)
dataset = f'Conv1d-base-w{window}-conv3'
dataset = f'Conv1d-base-w{window}-{filter}'
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}'))
os.makedirs('{}/{}/{}_{}-w{}'.format(self.name, self.loader.target, dataset, filter, window), exist_ok=True)
os.makedirs('{}/{}/{}-w{}'.format(self.name, self.loader.target, dataset, window), exist_ok=True)
X_train, X_test = X_conv1d[train_index], X_conv1d[test_index]
Y_train, Y_test = Y_conv1d[train_index], Y_conv1d[test_index]
G_train, G_test = G_conv1d[train_index], G_conv1d[test_index]
@ -556,7 +556,7 @@ class eNoseTrainer:
self.bar.update()
continue
model_file = '{}/{}/{}_{}-w{}/{}'.format(self.name, self.loader.target, dataset, filter, window, model_id )
model_file = '{}/{}/{}-w{}/{}'.format(self.name, self.loader.target, dataset, window, model_id )
X_train_sample, _, Y_train_sample, _ = train_test_split(X_train, Y_train, stratify=G_train, train_size=0.8*sample_size / len(X_train), random_state=get_seed())
X_test_sample, _, Y_test_sample, _ = train_test_split(X_test, Y_test, stratify=G_test, train_size=0.2*sample_size / len(X_test), random_state=get_seed())