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

checked masked cols after running

Browse source
This commit is contained in:
Tanushree Tunstall 2022-09-01 12:57:38 +01:00
parent f9129b9ebc
commit e03ce277b7
1 changed files with 1 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

16
scripts/ml/ml_functions/GetMLData.py
View file

@ -481,7 +481,7 @@ def getmldata(gene, drug
# mask the mcsm ligand affinity AND mcsm_na affinity columns where ligand distance > 10 # mask the mcsm ligand affinity AND mcsm_na affinity columns where ligand distance > 10
my_df_ml.loc[(my_df_ml['ligand_distance'] > 10), cols_to_mask] = 0 my_df_ml.loc[(my_df_ml['ligand_distance'] > 10), cols_to_mask] = 0
#mask_check = my_df_ml[['mutationinformation', 'ligand_distance'] + cols_to_mask] # Create names for masking columns
mask_check_cols = ['mutationinformation', 'ligand_distance'] + cols_to_mask mask_check_cols = ['mutationinformation', 'ligand_distance'] + cols_to_mask
#--------------------------- #---------------------------
@ -502,24 +502,10 @@ def getmldata(gene, drug
#mask_check = my_df_ml[['mutationinformation', 'ligand_distance'] + cols_to_mask + add_cols_mask] #mask_check = my_df_ml[['mutationinformation', 'ligand_distance'] + cols_to_mask + add_cols_mask]
mask_check_cols = mask_check_cols + add_cols_mask mask_check_cols = mask_check_cols + add_cols_mask
# if gene.lower() in geneL_na_ppi2:
# #---------------------------
# # RPOB: mask ppi2 + na + lig cols
# #---------------------------
# mask_check = my_df_ml[['mutationinformation',
# 'ligand_distance', 'ligand_affinity_change',
# 'nca_distance','mcsm_na_affinity',
# 'mcsm_ppi2_affinity','interface_dist']]
# GET mask data # GET mask data
mask_check = my_df_ml[mask_check_cols] mask_check = my_df_ml[mask_check_cols]
# sanity check: check script SANITY_CHECK_mask.py
if write_maskfile: if write_maskfile:
# write mask file for sanity check
#mask_check.sort_values(by = ['ligand_distance'], ascending = True, inplace = True)
mask_check.to_csv(outdir_ml + gene.lower() + '_mask_check.csv') mask_check.to_csv(outdir_ml + gene.lower() + '_mask_check.csv')
############################################################################### ###############################################################################