adding other split types on ml_iterator

2022-07-27 15:58:15 +01:00 · 2022-07-27 15:58:15 +01:00 · c32005c99c
commit c32005c99c
parent 63c8876764
1 changed files with 5 additions and 4 deletions
--- a/scripts/ml/ml_iterator.py
+++ b/scripts/ml/ml_iterator.py
@ -50,11 +50,12 @@ ml_gene_drugD = {'pncA'   : 'pyrazinamide'
 gene_dataD={}
 split_types = ['70_30',
               '80_20',
-               'sl'
-               ]
+               'sl',
+               'rt',
+               'none_bts']
+
 split_data_types = ['actual',
-                    'complete'
-                    ]
+                    'complete']

 for gene, drug in ml_gene_drugD.items():
    print ('\nGene:', gene