Add Working Milestone with Initial Results and Model Inference (#82)

* wip: add function to create stratified train-test split from STFT data

* feat(src): implement working function for dataset B to create ready data from STFT files stft_files and add setup.py for package configuration

* feat(notebook): Update variable names for clarity, remove unused imports, and streamline data processing. Implement data concatenation using pandas concat for efficiency. Add validation steps for Dataset B and improve model training consistency across sensors.

* fix(.gitignore): add rule to ignore egg-info directories and ensure proper formatting

* docs(README): add instructions for running stft.ipynb notebook

* feat(notebook): Add evaluation metrics and confusion matrix visualizations for model predictions on Dataset B. Remove commented-out code and integrate data preparation using create_ready_data function.

---------

Co-authored-by: nuluh <dam.ar@outlook.com>
This commit was merged in pull request #82.
This commit is contained in:
Rifqi D. Panuluh
2025-05-24 01:30:10 +07:00
committed by GitHub
parent a32415cebf
commit d151062115
7 changed files with 305 additions and 132 deletions

3
.gitignore vendored
View File

@@ -1,4 +1,5 @@
# Ignore CSV files in the data directory and all its subdirectories
data/**/*.csv
.venv/
*.pyc
*.pyc
*.egg-info/