From 685250de924e409d38c8158c1898423903fdc3ca Mon Sep 17 00:00:00 2001
From: Default user <default@maastrichtuniversity.nl>
Date: Fri, 8 Dec 2023 13:13:50 +0000
Subject: [PATCH] comment

---
 .gitignore               | 1 +
 src/train_opentargets.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 87acb2e..f42aa48 100644
--- a/.gitignore
+++ b/.gitignore
@@ -11,3 +11,4 @@ tmp/
 __pycache__/
 .pytest_cache
 .ipynb_checkpoints/
+*.log
\ No newline at end of file
diff --git a/src/train_opentargets.py b/src/train_opentargets.py
index c657b0e..d7309ed 100644
--- a/src/train_opentargets.py
+++ b/src/train_opentargets.py
@@ -50,7 +50,7 @@ def ensembl_to_uniprot():
 
 
 def train_opentargets(input_dir, out_dir):
-    """Main function to orchestrate the extraction and saving process."""
+    """Train the model using opentargets data."""
     os.makedirs(out_dir, exist_ok=True)
     known_drug_targets = []
 
@@ -82,6 +82,7 @@ def train_opentargets(input_dir, out_dir):
 
 
 def train_drugbank():
+    """Train the model using drugbank data."""
     df_known_dt = "data/drugbank/DB_DTI_4vectordb.csv"
     out_dir = "data/drugbank"