From 28b1123ae41e2bd7568a4454e5977ccf8cf776ec Mon Sep 17 00:00:00 2001 From: Israel Figueroa Date: Thu, 13 Mar 2025 23:17:57 -0300 Subject: [PATCH] patch --- TrainerClass.py | 2 +- train_sequence.py | 10 ++++------ 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/TrainerClass.py b/TrainerClass.py index 446ba56..c10c3a1 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -175,7 +175,7 @@ class eNoseTrainer: param_dist = self.get_tunable_params(model) cv = StratifiedShuffleSplit(n_splits=int(1/(2*self.ratio))+1, test_size=self.ratio, random_state=get_seed()) - grid_search = GridSearchCV(estimator=model, param_grid=param_dist, scoring='neg_mean_squared_error', cv=cv, verbose=10, n_jobs=-1) + grid_search = GridSearchCV(estimator=model, param_grid=param_dist, scoring='neg_mean_squared_error', cv=cv, verbose=2, n_jobs=-1) grid_search.fit(X_train, Y_train) diff --git a/train_sequence.py b/train_sequence.py index da5dbd5..efdf634 100644 --- a/train_sequence.py +++ b/train_sequence.py @@ -9,19 +9,17 @@ target_variables=['C2H2', 'CH4', 'C3H6', 'CO', 'C2H6', 'C3H8', 'C2H4', 'H2', 'O2 eNoseLoader = GasSensorDataLoader("enose_dataset", threshold=0.85, source_channels=source_channels, target_list=target_variables, debug=False) eNose = eNoseTrainer(eNoseLoader, test_size=0.2) +eNoseLoader.target_list=['H2',] +eNose.fit() eNoseLoader.target_list=['C2H2',] eNose.fit() eNoseLoader.target_list=['CH4',] eNose.fit() -eNoseLoader.target_list=['C3H6',] -eNose.fit() -eNoseLoader.target_list=['CO',] +eNoseLoader.target_list=['C2H4',] eNose.fit() eNoseLoader.target_list=['C2H6',] eNose.fit() -eNoseLoader.target_list=['C3H8',] -eNose.fit() -eNoseLoader.target_list=['C2H2', 'CH4', 'C3H6', 'CO', 'C2H6',] +eNoseLoader.target_list=['H2', 'C2H2', 'CH4', 'C2H4', 'C2H6',] eNose.fit() eNose.wrap_and_save()