HRV-emotion package

HRV-emotion package including machine learning and statistical analysis

Dataset

Lab datasets collected from experiments

🚀Quick setup

# clone project
git clone https://github.com/DimensionSTP/hrv-emotion.git
cd hrv-emotion

# [OPTIONAL] create conda environment
conda create -n myenv python=3.8
conda activate myenv

# install requirements
pip install -r requirements.txt

ECG or rPPG signal to HRV

• rename ECG data files before preprocessing

python rename_thirty_dataset_files_in_directory.py

• rename rPPG data files before preprocessing

python rename_sl_rppg_dataset_files_in_directory.py

• signal to HRV

./scripts/signal_to_hrv.sh

Preprocess HRV

• file path setting, normalization, averaging, make test dataset, statistical_analysis

./scripts/preprocess_hrv.sh

• concat augmented dataset for machine learning train dataset

python concat_augmented_dataset.py

Machine Learning(HRV-emotion classifier)

• model tuning

python main.py mode=tune condition={arousal or valence}

• model training

python main.py mode=train condition={arousal or valence} is_tuned={bool}

• model testing

python main.py mode=test condition={arousal or valence} data_type={test data}

• all model tuning

./scripts/tune.sh

• all model training

./scripts/train.sh

• all model testing

./scripts/test.sh

If you want to change main config, use --config-name={config_name}.

Also, you can use --multirun option.

You can set additional arguments through the command line.