Update README.md
Browse files
README.md
CHANGED
@@ -5,7 +5,7 @@ tags:
|
|
5 |
# Madrigal
|
6 |
|
7 |
Madrigal is an open-source model for predicting drug combination outcomes from multimodal preclinical data. This repository provides the weights of the model. Specifically, weights of the model after the following two stages are provided:
|
8 |
-
- Modality alignment: `pretrain/
|
9 |
- DDI finetuning (on DrugBank DDI dataset, initialized with the above modality alignment weights): `DrugBank/split_by_pairs/{run_name}/checkpoint_700.pt`
|
10 |
|
11 |
Note that the five `run_name` folders are the same model trained with different seeds, stopped at the same epoch.
|
|
|
5 |
# Madrigal
|
6 |
|
7 |
Madrigal is an open-source model for predicting drug combination outcomes from multimodal preclinical data. This repository provides the weights of the model. Specifically, weights of the model after the following two stages are provided:
|
8 |
+
- Modality alignment: `pretrain/checkpoint_1000.pt`
|
9 |
- DDI finetuning (on DrugBank DDI dataset, initialized with the above modality alignment weights): `DrugBank/split_by_pairs/{run_name}/checkpoint_700.pt`
|
10 |
|
11 |
Note that the five `run_name` folders are the same model trained with different seeds, stopped at the same epoch.
|