-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpredict_drag_drop.bat
More file actions
34 lines (28 loc) · 980 Bytes
/
predict_drag_drop.bat
File metadata and controls
34 lines (28 loc) · 980 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
@echo off
REM EEG Classification - Drag and Drop Prediction Tool
REM Drag EDF files onto this batch file to get predictions
echo EEG Classification - Drag and Drop Tool
echo ==========================================
REM Check if files were dragged onto the script
if "%~1"=="" (
echo No files were dragged onto this script.
echo Please drag one or more .edf files onto this batch file.
pause
exit /b 1
)
REM Set default model path (update this to match your setup)
set MODEL_PATH=models\eeg_classify_cnn_v1.h5
set SCALER_PATH=models\scaler_cnn.pkl
REM Check if model exists
if not exist "%MODEL_PATH%" (
echo Error: Model file %MODEL_PATH% not found.
echo Please make sure you have trained a model first.
pause
exit /b 1
)
REM Run the prediction script with all dragged files
echo Running predictions on dragged files...
python 5_predict.py --model_path "%MODEL_PATH%" --scaler_path "%SCALER_PATH%" %*
echo.
echo Predictions completed!
pause