Table - PMC

Skip to main content

View full-text article in PMC

. 2023 Apr 14;23(8):3984. doi: 10.3390/s23083984

Algorithm 1 Federated learning framework for stress detection

1:
$D_{s}$ (WESAD dataset)
2:
$D_{p}$ (SMOTE and Normalization)
3:
function global model upgrade
4:
weight initialization $w_{0}$
5:
for ( $t i = 1$ ) to T do
6:
for ( $c i = 1$ ) to C do
7:
$w_{t i + 1}^{c i}$ = local model upgrade(ci, $w_{t i}$ )
8:
$w_{t i + 1} = \sum_{c i = 1}^{C}$ $w_{D_{s}}^{c i}$ * $w_{t i + 1}^{c i}$
9:
function local model upgrade (ci, $w_{t i}$ )
10:
for ( $E p o c h = 1$ ) to $E_{p}$ do
11:
for ( $B_{s} \in B a t c h S i z e$ ) do
12:
$w = w - ▽ F_{i} (w)$
13:
return w