diff --git a/train_sequence.py b/train_sequence.py index fd70f07..da5dbd5 100644 --- a/train_sequence.py +++ b/train_sequence.py @@ -8,7 +8,7 @@ source_channels=["MQ 8", "MQ 9", "MQ 135", "TGS 813", "TGS 821", "TGS 2600", "TG 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.5) +eNose = eNoseTrainer(eNoseLoader, test_size=0.2) eNoseLoader.target_list=['C2H2',] eNose.fit() eNoseLoader.target_list=['CH4',]