dbroustail commited on
Commit
32e065b
·
verified ·
1 Parent(s): a2186eb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +75 -1
README.md CHANGED
@@ -20,13 +20,20 @@ datasets:
20
  - TUAB
21
  - APAVA
22
  - TDBrain
 
 
 
 
23
  metrics:
24
  - balanced_accuracy
25
  - roc_auc
26
  - pr_auc
 
 
 
27
  thumbnail: https://raw.githubusercontent.com/pulp-bio/BioFoundation/refs/heads/main/docs/model/logo/LuMamba_logo.png
28
  model-index:
29
- - name: LuMamba-Tiny
30
  results:
31
  - task:
32
  type: time-series-classification
@@ -70,6 +77,73 @@ model-index:
70
  - type: pr_auc
71
  value: 0.960
72
  name: AUC-PR
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
73
  ---
74
  <div align="center">
75
  <img src="https://raw.githubusercontent.com/pulp-bio/BioFoundation/refs/heads/main/docs/model/logo/LuMamba_logo.png" alt="LuMamba Logo" width="800"/>
 
20
  - TUAB
21
  - APAVA
22
  - TDBrain
23
+ - MoBI
24
+ - SEED-V
25
+ - Mumtaz2016
26
+ - MODMA
27
  metrics:
28
  - balanced_accuracy
29
  - roc_auc
30
  - pr_auc
31
+ - r2
32
+ - pearson_r
33
+ - cohen_kappa
34
  thumbnail: https://raw.githubusercontent.com/pulp-bio/BioFoundation/refs/heads/main/docs/model/logo/LuMamba_logo.png
35
  model-index:
36
+ - name: LuMamba-Tiny (LeJEPA-reconstruction pre-training)
37
  results:
38
  - task:
39
  type: time-series-classification
 
77
  - type: pr_auc
78
  value: 0.960
79
  name: AUC-PR
80
+ - task:
81
+ type: time-series-classification
82
+ name: Major Depressive Disorder Detection
83
+ dataset:
84
+ type: Mumtaz2016
85
+ name: Mumtaz2016
86
+ metrics:
87
+ - type: roc_auc
88
+ value: 0.931
89
+ name: AUROC
90
+ - type: pr_auc
91
+ value: 0.952
92
+ name: AUC-PR
93
+ - name: LuMamba-Tiny (Reconstruction-only pre-training)
94
+ results:
95
+ - task:
96
+ type: time-series-classification
97
+ name: EEG Slowing Event and Seizure Detection
98
+ dataset:
99
+ type: TUSL
100
+ name: TUH EEG Slowing Corpus (TUSL)
101
+ metrics:
102
+ - type: roc_auc
103
+ value: 0.708
104
+ name: AUROC
105
+ - type: pr_auc
106
+ value: 0.289
107
+ name: AUC-PR
108
+ - task:
109
+ type: time-series-classification
110
+ name: EEG Artifact Detection
111
+ dataset:
112
+ type: TUAR
113
+ name: TUH EEG Artifact Corpus (TUAR)
114
+ metrics:
115
+ - type: roc_auc
116
+ value: 0.914
117
+ name: AUROC
118
+ - type: pr_auc
119
+ value: 0.510
120
+ name: AUC-PR
121
+ - task:
122
+ type: time-series-classification
123
+ name: Gait Prediction Regression
124
+ dataset:
125
+ type: MoBI
126
+ name: MoBI
127
+ metrics:
128
+ - type: r2
129
+ value: 0.116
130
+ name: R squared
131
+ - type: rmse
132
+ value: 0.1482
133
+ name: Root Mean Squared Error
134
+ - task:
135
+ type: time-series-classification
136
+ name: 5-class Emotion Detection
137
+ dataset:
138
+ type: SEED-V
139
+ name: SEED-V
140
+ metrics:
141
+ - type: balanced_accuracy
142
+ value: 0.350
143
+ name: Balanced Accuracy (%)
144
+ - type: cohen_kappa
145
+ value: 0.191
146
+ name: Cohen's Kappa
147
  ---
148
  <div align="center">
149
  <img src="https://raw.githubusercontent.com/pulp-bio/BioFoundation/refs/heads/main/docs/model/logo/LuMamba_logo.png" alt="LuMamba Logo" width="800"/>