Skip to content

Commit c48a894

Browse files
committed
added details of the other models
1 parent 53bbe8e commit c48a894

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

README.md

+3-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,9 @@ The code relies primarily on the [PyKEEN](https://github.com/pykeen/pykeen) pack
3535

3636
A pretrained version of the [TransE](https://proceedings.neurips.cc/paper/2013/file/1cecc7a77928ca8133fa24680a88d2f9-Paper.pdf) model is provided in the `artifacts` directory. This version is automatically used in the notebooks.
3737

38-
However, code is provided to retrain the model or swap it out for a different one in the `train.py` file in the `src` directory. Please use this function to retrain the other models evaluated in the study.
38+
Due to the size of weight files for the other models used in the study, they were unable to be included in main repository. However, the files are available for download as part of the release of the repository [here](https://github.com/AstraZeneca/biomedical-kg-topological-imbalance/releases/tag/0.2.0). Once the files have been downloaded, please place them in the `artifacts` directory if you wish to use them.
39+
40+
Note that the code is provided to retrain any of the models in the `train.py` file in the `src` directory.
3941

4042
## Citation
4143

0 commit comments

Comments
 (0)