From 7eef46391579237d77dad83cadf0f555e6f35c3f Mon Sep 17 00:00:00 2001
From: Tanushree Tunstall <tanu@tunstall.in>
Date: Wed, 29 Jun 2022 22:20:23 +0100
Subject: [PATCH] TEMP -> ml_iterator

---
 scripts/ml/{TEMP => ml_iterator.py} | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)
 rename scripts/ml/{TEMP => ml_iterator.py} (93%)
 mode change 100644 => 100755

diff --git a/scripts/ml/TEMP b/scripts/ml/ml_iterator.py
old mode 100644
new mode 100755
similarity index 93%
rename from scripts/ml/TEMP
rename to scripts/ml/ml_iterator.py
index ebb4549..a9be2b4
--- a/scripts/ml/TEMP
+++ b/scripts/ml/ml_iterator.py
@@ -32,13 +32,13 @@ combined_model_paramD = {'data_combined_model'   : False
 #ml_genes = ["pncA", "embB", "katG", "rpoB", "gid"]
 
 ml_gene_drugD = {'pncA'   : 'pyrazinamide'
-                 # , 'embB' : 'ethambutol'
-                 # , 'katG' : 'isoniazid'
-                 # , 'rpoB' : 'rifampicin'
-                 # , 'gid'  : 'streptomycin'
+                 , 'embB' : 'ethambutol'
+                 , 'katG' : 'isoniazid'
+                 , 'rpoB' : 'rifampicin'
+                 , 'gid'  : 'streptomycin'
                  }
 gene_dataD={}
-split_types = ['70_30', '80_20']
+split_types = ['70_30', '80_20', 'sl']
 split_data_types = ['actual', 'complete']
 
 for gene, drug in ml_gene_drugD.items():
@@ -98,5 +98,5 @@ for gene, drug in ml_gene_drugD.items():
                 out_wf= pd.concat(mmDD, ignore_index = True)
             
             out_wf_f = out_wf.sort_values(by = ['resampling', 'source_data', 'MCC'], ascending = [True, True, False], inplace = False)
-            out_wf_f.to_csv(('/tmp/'+out_filename), index = False)
+            out_wf_f.to_csv(('/home/tanu/git/Data/ml_combined/genes/'+out_filename), index = False)