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

preparing to rerun ML iterator

Browse source
This commit is contained in:
Tanushree Tunstall 2022-09-01 13:41:02 +01:00
parent 56b71c6ca2
commit 2bf91681b4
2 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

1
scripts/ml/ml_functions/TEST_GetMLData.py
View file

@ -5,6 +5,7 @@ Created on Thu Sep 1 12:22:27 2022
@author: tanu @author: tanu
""" """
getmldata(gene = "alr" getmldata(gene = "alr"
, drug = "cycloserine" , drug = "cycloserine"
, data_combined_model = False , data_combined_model = False

13
scripts/ml/ml_iterator.py
View file

@ -40,10 +40,9 @@ gene_model_paramD = {'data_combined_model' : False
, 'write_outfile' : False } , 'write_outfile' : False }
############################################################################### ###############################################################################
#ml_genes = ["pncA", "embB", "katG", "rpoB", "gid"] #ml_genes = ["pncA", "embB", "katG", "rpoB", "gid"]
ml_gene_drugD = { ml_gene_drugD = {
#'pncA' : 'pyrazinamide' 'pncA' : 'pyrazinamide'
'embB' : 'ethambutol' , 'embB' : 'ethambutol'
, 'katG' : 'isoniazid' , 'katG' : 'isoniazid'
, 'rpoB' : 'rifampicin' , 'rpoB' : 'rifampicin'
, 'gid' : 'streptomycin' , 'gid' : 'streptomycin'
@ -53,13 +52,13 @@ split_types = [
'70_30', '70_30',
'80_20', '80_20',
'sl', 'sl',
#'rt', 'rt',
#'none_bts' 'none_bts'
] ]
split_data_types = [ split_data_types = [
'actual', #'actual',
#'complete' 'complete'
] ]
for gene, drug in ml_gene_drugD.items(): for gene, drug in ml_gene_drugD.items():