From c32005c99c6cd223e905204dbe25c0cbd48a195f Mon Sep 17 00:00:00 2001 From: Tanushree Tunstall Date: Wed, 27 Jul 2022 15:58:15 +0100 Subject: [PATCH] adding other split types on ml_iterator --- scripts/ml/ml_iterator.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/scripts/ml/ml_iterator.py b/scripts/ml/ml_iterator.py index 6daf527..ea9002c 100755 --- 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