tanu/LSHTM_analysis
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

adding other split types on ml_iterator

Browse source
This commit is contained in:
Tanushree Tunstall 2022-07-27 15:58:15 +01:00
parent 63c8876764
commit c32005c99c
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
scripts/ml/ml_iterator.py
View file

@ -50,11 +50,12 @@ ml_gene_drugD = {'pncA' : 'pyrazinamide'
gene_dataD={} gene_dataD={}
split_types = ['70_30', split_types = ['70_30',
'80_20', '80_20',
'sl' 'sl',
] 'rt',
'none_bts']
split_data_types = ['actual', split_data_types = ['actual',
'complete' 'complete']
]
for gene, drug in ml_gene_drugD.items(): for gene, drug in ml_gene_drugD.items():
print ('\nGene:', gene print ('\nGene:', gene