seizuretransformer 0.0.1

SeizureTransformer EEG model in Rust (Burn + wgpu)
Documentation 
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

use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SeizureTransformerConfig {
    pub in_channels: usize,
    pub in_samples: usize,
    pub dim_feedforward: usize,
    pub num_layers: usize,
    pub num_heads: usize,
    pub drop_rate: f64,
    pub max_pos_len: usize,
}

impl Default for SeizureTransformerConfig {
    fn default() -> Self {
        Self {
            in_channels: 18,
            in_samples: 15360,
            dim_feedforward: 2048,
            num_layers: 8,
            num_heads: 4,
            drop_rate: 0.1,
            max_pos_len: 6000,
        }
    }
}