luanbei commited on
Commit
6619cb8
Β·
verified Β·
1 Parent(s): b3fa33b

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +6 -3
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 scripts/predict_binary_hierarchy.py \
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 scripts/predict_binary.py \
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 scripts/predict_hierarchy.py \
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 \