Skip to content

Commit 0e3ec9a

Browse files
authored
Update README.md
1 parent 4e2cd06 commit 0e3ec9a

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

README.md

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,3 +63,27 @@ pip install torch numpy scikit-learn json argparse
6363
```bash
6464
python train_model_paras_combine.py
6565
```
66+
run_immudef.py — Immune Defense Score (IS) Computation Pipeline
67+
68+
This script implements the complete computational workflow for deriving the Immune Defense Score (IS) based on QImmuDef-VAE latent representations and ensemble-model–trained reference samples. It is designed for reproducible scoring of immune defense capacity for both healthy controls and disease cohorts.
69+
70+
## DImmuScore calculation
71+
72+
### Command-Line Usage
73+
```bash
74+
python run_immudef.py \
75+
--input data/expression_matrix.csv \
76+
--vae_model models/QImmuDefVAE.pt \
77+
--ensemble_model models/ensemble.pkl \
78+
--gene_list data/immune_gene_list.txt \
79+
--outdir results/
80+
```
81+
### Required Input Files
82+
Argument Description
83+
```bash
84+
--input Expression matrix (.csv or .tsv). Rows = samples,
85+
--vae_model Pretrained QImmuDef-VAE weights (.pt).
86+
--ensemble_model Trained ensemble classifier combining SVM, XGBoost and Logistic Regression (.pkl).
87+
--gene_list List of immune-related genes used during VAE training.
88+
--outdir Output folder where all results will be saved.
89+
```

0 commit comments

Comments
 (0)