This script performs fatigue inference on user-provided EMG (Electromyography) signals using a pre-trained LightGBM model.
ROC Curve
Features
Predictions by the model
- Python 3.x installed
- Required Python packages installed (install with
pip install -r requirements.txt
)
-
Clone the repository to your local machine:
git clone https://github.com/your-username/sEMG-Automatic-Detection.git
- inference.py: The main script for performing fatigue inference.
- process_data.py: Module containing functions for processing EMG data.
- models/: Directory containing the pre-trained LightGBM model file.
- selected_emgs/: Directory where user EMG files can be stored.
- Ensure that your EMG signal files have a
.txt
extension and are formatted correctly. - The script uses a pre-trained LightGBM model located in the
models/
directory. - Feel free to modify the script or add more features based on your needs.
-
Open a terminal and navigate to the project directory.
-
Run the inference.py script:
python inference.py path/to/emg/folder
Replace path/to/emg/folder with the path to the folder containing your EMG signal files.
-
The script will list the available EMG files in the specified folder. Enter the number corresponding to the file you want to evaluate.
-
The script will process the selected EMG file, perform fatigue inference, and display the results.