Commit 224875dd authored by ADRIAN  AYUSO MUNOZ's avatar ADRIAN AYUSO MUNOZ

Fix README.md

parent 3de5f8ba
......@@ -29,9 +29,9 @@ drugs and diseases (edge inference) or predict the type of edge (drug) between t
Four models will be generated using different libraries to compare them and use the one, or a combination of models,
that fits the project's necessities the most. All the models have encoder-decoder (GraphSAGE-Dot product) architectures,
the use of different architectures will be studied in the future. These are the generated models and their status:
- 🟢 **DeepSnap:** Passed validation tests, proved to be functional and works well.
- 🟡 **DGL:** Halted before validation tests, it seems that there is data leakage between the train and the validation set.
- 🟡 **DGL Classification:** Halted before validation tests, need to work on memory management.
- 🟢 **DeepSnap:** Passed verification tests, proved to be functional and works well.
- 🟡 **DGL:** Halted before verification tests, it seems that there is data leakage between the train and the validation set.
- 🟡 **DGL Classification:** Halted before verification tests, need to work on memory management.
- 🟠 **Pytorch Geometric:** Could not develop a functional GNN model.
### Symbol legend:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment