From 80b4a1850c6c2ff211ed58b85f0b51782ba10ecb Mon Sep 17 00:00:00 2001 From: Tanushree Tunstall Date: Thu, 1 Sep 2022 16:23:57 +0100 Subject: [PATCH] tested to make sure the cols added dont break code --- scripts/ml/ml_functions/GetMLData.py | 5 +++-- scripts/ml/ml_functions/TEST_GetMLData.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/scripts/ml/ml_functions/GetMLData.py b/scripts/ml/ml_functions/GetMLData.py index bd2d703..971fe6c 100755 --- a/scripts/ml/ml_functions/GetMLData.py +++ b/scripts/ml/ml_functions/GetMLData.py @@ -657,7 +657,8 @@ def getmldata(gene, drug all_featuresN2 = all_featuresN + ['dst', 'dst_mode'] all_training_df = my_df_ml[all_featuresN2] - print('\nAll feature names:', all_featuresN2) + print('\nLength of all ML feature names:', len(all_featuresN2) + , '\nAll feature names:', all_featuresN2) #################################################################### #========================================================================== @@ -670,7 +671,7 @@ def getmldata(gene, drug #========================================================================== if write_outfile: print('\nPASS: and now writing processed file for ml:', outFile_ml) - #all_training_df.to_csv(outFile_ml, index = False) + all_training_df.to_csv(outFile_ml, index = False) else: print('\nPASS: But NOT writing processed file') #========================================================================== diff --git a/scripts/ml/ml_functions/TEST_GetMLData.py b/scripts/ml/ml_functions/TEST_GetMLData.py index be1c14b..3222eec 100644 --- a/scripts/ml/ml_functions/TEST_GetMLData.py +++ b/scripts/ml/ml_functions/TEST_GetMLData.py @@ -12,7 +12,7 @@ getmldata(gene = "alr" , use_or = False , omit_all_genomic_features = False , write_maskfile = True - , write_outfile = False) + , write_outfile = True) getmldata(gene = "embB"