Research Article

Stress Classification Using Brain Signals Based on LSTM Network

Table 2

Parameters for LSTM models chosen after hyperparameter tuning.

ParameterValues

Number of input features20
Number of output features1
Number of LSTM layers2
Number of hidden units in LSTM layers8 and 16 (only for two-layer LSTM)
Activation functionSigmoid
OptimizerAdam
Loss functionBinary cross entropy
Batch size32
Window size20
Epochs100
Dropout value0.2