Update README.md
Browse files
README.md
CHANGED
@@ -2,12 +2,9 @@
|
|
2 |
base_model:
|
3 |
- autogluon/chronos-t5-large
|
4 |
---
|
5 |
-
|
6 |
-
|
7 |
-
<!-- Provide a quick summary of what the model is/does. -->
|
8 |
-
|
9 |
-
This modelcard aims to be a base template for new models. It has been generated using [this raw template](https://github.com/huggingface/huggingface_hub/blob/main/src/huggingface_hub/templates/modelcard_template.md?plain=1).
|
10 |
|
|
|
11 |
## Model Details
|
12 |
|
13 |
### Model Description
|
@@ -15,62 +12,15 @@ This modelcard aims to be a base template for new models. It has been generated
|
|
15 |
<!-- Provide a longer summary of what this model is. -->
|
16 |
|
17 |
- **Model type:** T5-large
|
18 |
-
- **Language(s) (NLP):** [More Information Needed]
|
19 |
- **License:** [More Information Needed]
|
20 |
-
- **Finetuned from model [optional]:** Chronos
|
21 |
-
|
22 |
-
### Model Sources [optional]
|
23 |
-
|
24 |
-
<!-- Provide the basic links for the model. -->
|
25 |
-
|
26 |
-
- **Repository:** [More Information Needed]
|
27 |
-
- **Paper [optional]:** [More Information Needed]
|
28 |
-
- **Demo [optional]:** [More Information Needed]
|
29 |
-
|
30 |
-
## Uses
|
31 |
-
|
32 |
-
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
33 |
-
|
34 |
-
### Direct Use
|
35 |
-
|
36 |
-
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
37 |
-
|
38 |
-
[More Information Needed]
|
39 |
-
|
40 |
-
### Downstream Use [optional]
|
41 |
-
|
42 |
-
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
|
43 |
-
|
44 |
-
[More Information Needed]
|
45 |
-
|
46 |
-
### Out-of-Scope Use
|
47 |
-
|
48 |
-
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
49 |
-
|
50 |
-
[More Information Needed]
|
51 |
-
|
52 |
-
## Bias, Risks, and Limitations
|
53 |
-
|
54 |
-
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
55 |
-
|
56 |
-
[More Information Needed]
|
57 |
-
|
58 |
-
### Recommendations
|
59 |
-
|
60 |
-
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
61 |
-
|
62 |
-
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
63 |
|
64 |
## How to Get Started with the Model
|
65 |
|
66 |
-
|
67 |
-
|
68 |
-
[More Information Needed]
|
69 |
-
|
70 |
-
## Training Details
|
71 |
|
72 |
-
### Training Data
|
|
|
73 |
|
74 |
-
|
75 |
|
76 |
-
[More Information Needed]
|
|
|
2 |
base_model:
|
3 |
- autogluon/chronos-t5-large
|
4 |
---
|
5 |
+
General-Purpose Brain Foundation Models is trained on electroencephalogram signal data.
|
|
|
|
|
|
|
|
|
6 |
|
7 |
+
[[paper]()]
|
8 |
## Model Details
|
9 |
|
10 |
### Model Description
|
|
|
12 |
<!-- Provide a longer summary of what this model is. -->
|
13 |
|
14 |
- **Model type:** T5-large
|
|
|
15 |
- **License:** [More Information Needed]
|
16 |
+
- **Finetuned from model [optional]:** Chronos-t5-large
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
|
18 |
## How to Get Started with the Model
|
19 |
|
20 |
+
Here is [colab](https://colab.research.google.com/drive/1Dy6oUJeYuoi0cAo9imlfA2gZgueMun9h?usp=sharing) notebook for inference of Moabb dataset.
|
|
|
|
|
|
|
|
|
21 |
|
22 |
+
### Training Data:
|
23 |
+
Training data used was NMT EEG dataset. NMT is an open-source, annotated dataset comprising healthy and pathological EEG recordings. It consists of 2,417 recordings from unique participants, providing multichannel EEG data along with labels indicating the participants' pathological state, classified as normal or abnormal. Each EEG channel is treated as an independent time series, which is further divided into two segments: a context window for conditioning and a prediction target window.
|
24 |
|
25 |
+
Hyperparameters are the same as used in Chronos paper.
|
26 |
|
|