Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -22,6 +22,8 @@ This repository contains the trained PACMT model files and taxonomy resources. T
|
|
| 22 |
https://github.com/luanbei/PACMT
|
| 23 |
```
|
| 24 |
|
|
|
|
|
|
|
| 25 |
## Model description
|
| 26 |
|
| 27 |
PACMT uses a two-stage serial workflow:
|
|
@@ -48,6 +50,7 @@ PACMT/
|
|
| 48 |
β βββ bert_padding.py
|
| 49 |
β βββ flash_attn_triton.py
|
| 50 |
βββ binary_model/
|
|
|
|
| 51 |
β βββ pytorch_model.bin
|
| 52 |
β βββ head_config.json
|
| 53 |
β βββ tokenizer.json
|
|
@@ -126,7 +129,7 @@ models/
|
|
| 126 |
The complete two-stage PACMT workflow first performs binary viral screening and then applies hierarchical taxonomic classification to sequences predicted as viral.
|
| 127 |
|
| 128 |
```bash
|
| 129 |
-
python
|
| 130 |
--backbone_dir models/backbone \
|
| 131 |
--binary_ckpt_dir models/binary_model \
|
| 132 |
--hierarchy_ckpt_dir models/hierarchy_model \
|
|
@@ -154,7 +157,7 @@ For FASTA input, replace the CSV input arguments with:
|
|
| 154 |
## Binary viral screening only
|
| 155 |
|
| 156 |
```bash
|
| 157 |
-
python
|
| 158 |
--backbone_dir models/backbone \
|
| 159 |
--ckpt_dir models/binary_model \
|
| 160 |
--input_csv examples/example.csv \
|
|
@@ -173,7 +176,7 @@ python scripts/predict_binary.py \
|
|
| 173 |
## Hierarchical classification only
|
| 174 |
|
| 175 |
```bash
|
| 176 |
-
python
|
| 177 |
--backbone_dir models/backbone \
|
| 178 |
--ckpt_dir models/hierarchy_model \
|
| 179 |
--mapping_csv models/taxonomy/label_taxonomy_mapping.csv \
|
|
|
|
| 22 |
https://github.com/luanbei/PACMT
|
| 23 |
```
|
| 24 |
|
| 25 |
+
Use the current scripts from the GitHub repository's `script/` directory. The three prediction scripts require `pacmt_retrained_common.py` in that same directory.
|
| 26 |
+
|
| 27 |
## Model description
|
| 28 |
|
| 29 |
PACMT uses a two-stage serial workflow:
|
|
|
|
| 50 |
β βββ bert_padding.py
|
| 51 |
β βββ flash_attn_triton.py
|
| 52 |
βββ binary_model/
|
| 53 |
+
β βββ config.json
|
| 54 |
β βββ pytorch_model.bin
|
| 55 |
β βββ head_config.json
|
| 56 |
β βββ tokenizer.json
|
|
|
|
| 129 |
The complete two-stage PACMT workflow first performs binary viral screening and then applies hierarchical taxonomic classification to sequences predicted as viral.
|
| 130 |
|
| 131 |
```bash
|
| 132 |
+
python script/predict_bina_hiera.py \
|
| 133 |
--backbone_dir models/backbone \
|
| 134 |
--binary_ckpt_dir models/binary_model \
|
| 135 |
--hierarchy_ckpt_dir models/hierarchy_model \
|
|
|
|
| 157 |
## Binary viral screening only
|
| 158 |
|
| 159 |
```bash
|
| 160 |
+
python script/predict_binary.py \
|
| 161 |
--backbone_dir models/backbone \
|
| 162 |
--ckpt_dir models/binary_model \
|
| 163 |
--input_csv examples/example.csv \
|
|
|
|
| 176 |
## Hierarchical classification only
|
| 177 |
|
| 178 |
```bash
|
| 179 |
+
python script/predict_hierarchy.py \
|
| 180 |
--backbone_dir models/backbone \
|
| 181 |
--ckpt_dir models/hierarchy_model \
|
| 182 |
--mapping_csv models/taxonomy/label_taxonomy_mapping.csv \
|