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

ml_iterator tweaks

Browse source
This commit is contained in:
Tanushree Tunstall 2022-07-28 13:19:46 +01:00
parent 584b866f0b
commit 90b9477520
1 changed files with 17 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

25
scripts/ml/ml_iterator.py
View file

@ -41,21 +41,26 @@ gene_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'
ml_gene_drugD = {
#'pncA' : 'pyrazinamide'
'embB' : 'ethambutol'
, 'katG' : 'isoniazid'
, 'rpoB' : 'rifampicin'
, 'gid' : 'streptomycin'
}
gene_dataD={}
split_types = ['70_30',
split_types = [
'70_30',
'80_20',
'sl',
'rt',
'none_bts']
#'rt',
#'none_bts'
]
split_data_types = ['actual',
'complete']
split_data_types = [
'actual',
#'complete'
]
for gene, drug in ml_gene_drugD.items():
print ('\nGene:', gene