From 6c10776ea912682051e5f217d31267b27de5400b Mon Sep 17 00:00:00 2001
From: Tanushree Tunstall <tanu@tunstall.in>
Date: Sat, 25 Jun 2022 14:15:21 +0100
Subject: [PATCH] saving work

---
 scripts/ml/functions/GetMLData.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/scripts/ml/functions/GetMLData.py b/scripts/ml/functions/GetMLData.py
index cfd762a..f8155d9 100755
--- a/scripts/ml/functions/GetMLData.py
+++ b/scripts/ml/functions/GetMLData.py
@@ -609,8 +609,11 @@ def getmldata(gene, drug
 
     print('\n#################################################################'
           , '\nSUCCESS: Extacted training data for gene:', gene.lower()
-          , '\nDim of training_df:', all_training_df.shape
-          , '\nThis includes OR'
+          , '\nDim of training_df:', all_training_df.shape)
+    if use_or:
+        print('\nThis includes Odds Ratio')
+    else:
+        print('\nThis EXCLUDES Odds Ratio'
           , '\n###############################################################')
 
     #==========================================================================