optimised run_7030.py to generate ouput from dict now that the processfunction and parameter dicts have been added

This commit is contained in:
Tanushree Tunstall 2022-06-24 15:40:18 +01:00
parent 7dc7e25016
commit b37a950fec
12 changed files with 180 additions and 128408 deletions

View file

@ -7,21 +7,24 @@ Created on Fri Jun 24 11:07:05 2022
"""
import re
import argparse
import os, sys
###############################################################################
# gene = 'pncA'
# drug = 'pyrazinamide'
#total_mtblineage_uc = 8
#%% command line args: case sensitive
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument('-d', '--drug', help = 'drug name', default = 'pncA')
arg_parser.add_argument('-g', '--gene', help = 'gene name', default = 'pyrazinamide')
args = arg_parser.parse_args()
# #%% command line args: case sensitive
# arg_parser = argparse.ArgumentParser()
# arg_parser.add_argument('-d', '--drug', help = 'drug name', default = '')
# arg_parser.add_argument('-g', '--gene', help = 'gene name', default = '')
# args = arg_parser.parse_args()
drug = args.drug
gene = args.gene
# drug = args.drug
# gene = args.gene
###############################################################################
homedir = os.path.expanduser("~")
sys.path.append(homedir + '/git/LSHTM_analysis/scripts/ml')
###############################################################################
#==================
@ -79,7 +82,7 @@ mmD = MultModelsCl(input_df = X_smnc
#================
# MultModelsCl: WITH formatted output
#================
mmDF = MultModelsCl(input_df = X_smnc
mmDF3 = MultModelsCl(input_df = X_smnc
, target = y_smnc
, var_type = 'mixed'
, tts_split_type = tts_split_7030
@ -96,4 +99,4 @@ mmDF = MultModelsCl(input_df = X_smnc
# test function
#=================
# output from function call
ProcessMultModelCl(smnc_scores_mmD)
ProcessMultModelsCl(mmD)