Update README.md
Browse files
README.md
CHANGED
@@ -5,19 +5,23 @@ The AIDO.Tissue model is develpoed on spatial single-cell transcriptomic data. T
|
|
5 |
|
6 |
We evelauted the model on two spatial data task, including predicting niche label and cell density. The metrics are as below:
|
7 |
|
8 |
-
|
|
|
9 |
| -------- | ------- |
|
10 |
-
|
|
|
|
11 |
|
|
|
12 |
| Task | Mean absolute error | R square |
|
13 |
| -------- | ------- | ------- |
|
14 |
-
|
|
|
|
15 |
|
16 |
# Finetuning AIDO.Tissue for spatial single cell downstream tasks
|
17 |
We introduce how to finetune and evaluate our pre-trained AIDO.Tissue foundation models for downstream tasks. These tasks can be classified into the following categories:
|
18 |
|
19 |
-
* **
|
20 |
-
* **
|
21 |
|
22 |
Note: All the following scripts should be run under `ModelGenerator/`.
|
23 |
|
@@ -28,7 +32,7 @@ For each `.h5ad`, several obs attributes should be included to reprezent the spa
|
|
28 |
|
29 |
Note: the file `scRNA_genename_and_index.tsv` includes all the corresponding gene name and index in h5ad file.
|
30 |
|
31 |
-
##
|
32 |
### niche label type prediction
|
33 |
We fully finetune AIDO.Tissue for niche label type prediction.
|
34 |
|
@@ -56,7 +60,7 @@ CUDA_VISIBLE_DEVICES=6 nohup mgen test --config experiments/AIDO.Tissue/niche_ty
|
|
56 |
Note: `ckpt_path` is the finetuned checkpoint path.
|
57 |
|
58 |
|
59 |
-
##
|
60 |
|
61 |
### cell density prediction
|
62 |
|
|
|
5 |
|
6 |
We evelauted the model on two spatial data task, including predicting niche label and cell density. The metrics are as below:
|
7 |
|
8 |
+
niche label type prediction:
|
9 |
+
| Model | F1-score |
|
10 |
| -------- | ------- |
|
11 |
+
| AIDO.Tissue | 0.67 |
|
12 |
+
| Nicheformer | 0.50 |
|
13 |
|
14 |
+
cell density prediction:
|
15 |
| Task | Mean absolute error | R square |
|
16 |
| -------- | ------- | ------- |
|
17 |
+
| AIDO.Tissue | 4.44 | 0.55|
|
18 |
+
| Nicheformer | 7.08 | -0.07|
|
19 |
|
20 |
# Finetuning AIDO.Tissue for spatial single cell downstream tasks
|
21 |
We introduce how to finetune and evaluate our pre-trained AIDO.Tissue foundation models for downstream tasks. These tasks can be classified into the following categories:
|
22 |
|
23 |
+
* **Cell-level classification tasks**: niche label type prediction
|
24 |
+
* **Cell-level regression tasks**: cell density prediction
|
25 |
|
26 |
Note: All the following scripts should be run under `ModelGenerator/`.
|
27 |
|
|
|
32 |
|
33 |
Note: the file `scRNA_genename_and_index.tsv` includes all the corresponding gene name and index in h5ad file.
|
34 |
|
35 |
+
## Cell-level classification tasks
|
36 |
### niche label type prediction
|
37 |
We fully finetune AIDO.Tissue for niche label type prediction.
|
38 |
|
|
|
60 |
Note: `ckpt_path` is the finetuned checkpoint path.
|
61 |
|
62 |
|
63 |
+
## Cell-level regression tasks
|
64 |
|
65 |
### cell density prediction
|
66 |
|