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

patch

main
ifiguero 2025-03-09 04:47:45 -03:00
parent 8d3bc4ba2c
commit 431ef96b9a
3 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
LoaderClass.py
View File

@ -28,7 +28,6 @@ class GasSensorDataLoader:
self.source_channels = sorted(source_channels)
self.force_overwrite = force_overwrite
self.output_format = output_format.lower()
self.pics_folder = self.create_pics_folder()
self.logger = logging.getLogger("GasSensorDataLoader")
if self.debug:
@ -36,6 +35,8 @@ class GasSensorDataLoader:
else:
self.logger.setLevel(logging.INFO)
self.pics_folder = self.create_pics_folder()
if self.output_format not in ["png", "pdf"]:
raise ValueError("Invalid output format. Choose either 'png' or 'pdf'.")

BIN
enose_dataset.tgz 100644
View File

Binary file not shown.

10
train_sequence.py
View File

@ -7,15 +7,15 @@ warnings.filterwarnings("ignore")
source_channels=["MQ 8", "MQ 9", "MQ 135", "TGS 813", "TGS 821", "TGS 2600", "TGS 2602", "TGS 2611-0", "TGS 2612", "TGS 2620"]
#target_variables=['C2H2', 'CH4', 'C3H6', 'CO', 'C2H6', 'C3H8', 'C2H4', 'H2', 'O2']
#target_variables=['C2H2']
#eNoseLoaderC2H2 = GasSensorDataLoader("enose_dataset", threshold=0.85, source_channels=source_channels, target_list=target_variables, debug=False)
#eNoseC2H2 = eNoseTrainer(eNoseLoaderC2H2, splits=3, test_size=0.2, debug=True)
#eNoseC2H2.fit()
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.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=3, test_size=0.2, debug=True)
eNoseCH4 = eNoseTrainer(eNoseLoaderCH4, splits=7, test_size=0.2, debug=True)
eNoseCH4.fit()
#eNose.wrap_and_save()