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

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