main
ifiguero 2025-03-10 18:49:22 -03:00
parent f5bff5cd81
commit 4d2a32e75d
2 changed files with 23 additions and 5 deletions

View File

@ -173,7 +173,7 @@ class eNoseTrainer:
def train_and_score_model(self, model, X_train, X_test, y_train, y_test):
param_dist = self.get_tunable_params(model)
cv = ShuffleSplit(n_splits=self.splits//2, test_size=0.2, random_state=get_seed())
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.fit(X_train, y_train)
@ -204,7 +204,7 @@ class eNoseTrainer:
node = os.uname()[1]
X_xboost, Y_xboost, G_xboost = self.loader.load_dataset_xboost()
gss = GroupShuffleSplit(n_splits=self.splits, test_size=self.ratio, random_state=get_seed())
gss = StratifiedGroupKFold(n_splits=int(1/self.ratio), random_state=get_seed())
dataset = 'Tabular'
os.makedirs('{}/{}/{}'.format(self.name, self.target, dataset), exist_ok=True)

View File

@ -9,13 +9,31 @@ source_channels=["MQ 8", "MQ 9", "MQ 135", "TGS 813", "TGS 821", "TGS 2600", "TG
target_variables=['C2H2']
eNoseLoaderC2H2 = GasSensorDataLoader("enose_dataset", threshold=0.85, source_channels=source_channels, target_list=target_variables, debug=False)
eNoseC2H2 = eNoseTrainer(eNoseLoaderC2H2, splits=7, test_size=0.2, debug=True)
eNoseC2H2 = eNoseTrainer(eNoseLoaderC2H2, test_size=0.2, debug=True)
eNoseC2H2.fit()
target_variables=['CH4']
eNoseLoaderCH4 = GasSensorDataLoader("enose_dataset", threshold=0.85, source_channels=source_channels, target_list=target_variables, debug=False)
eNoseCH4 = eNoseTrainer(eNoseLoaderCH4, splits=7, test_size=0.2, debug=True)
eNoseCH4 = eNoseTrainer(eNoseLoaderCH4, test_size=0.2, debug=True)
eNoseCH4.fit()
target_variables=['C3H6']
eNoseLoaderC3H6 = GasSensorDataLoader("enose_dataset", threshold=0.85, source_channels=source_channels, target_list=target_variables, debug=False)
eNoseC3H6 = eNoseTrainer(eNoseLoaderC3H6, test_size=0.2, debug=True)
eNoseC3H6.fit()
target_variables=['C2H6']
eNoseLoaderC2H6 = GasSensorDataLoader("enose_dataset", threshold=0.85, source_channels=source_channels, target_list=target_variables, debug=False)
eNoseC2H6 = eNoseTrainer(eNoseLoaderC2H6, test_size=0.2, debug=True)
eNoseC2H6.fit()
target_variables=['H2']
eNoseLoaderH2 = GasSensorDataLoader("enose_dataset", threshold=0.85, source_channels=source_channels, target_list=target_variables, debug=False)
eNoseH2 = eNoseTrainer(eNoseLoaderH2, test_size=0.2, debug=True)
eNoseH2.fit()
#eNose.wrap_and_save()