gongjing commited on
Commit
d50445f
·
verified ·
1 Parent(s): e3c4731

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -7
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
- | Task | F1-score |
 
9
  | -------- | ------- |
10
- | niche label type prediction | 0.67 |
 
11
 
 
12
  | Task | Mean absolute error | R square |
13
  | -------- | ------- | ------- |
14
- | cell density prediction | 4.44 | 0.55|
 
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
- * **Sequence-level classification tasks**: niche label type prediction
20
- * **Sequence-level regression tasks**: cell density prediction
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
- ## Sequence-level classification tasks
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
- ## Sequence-level regression tasks
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