From 977c16425091b3af312d668bda439265c9c83927 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Thu, 19 Oct 2023 10:28:31 +0000 Subject: [PATCH] Skip merging the column named 'is_installed' --- 0_train/2_supGNN/merge_train_FE_and_train_supgnn_data.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/0_train/2_supGNN/merge_train_FE_and_train_supgnn_data.py b/0_train/2_supGNN/merge_train_FE_and_train_supgnn_data.py index 5c7ffcb..1c5daef 100644 --- a/0_train/2_supGNN/merge_train_FE_and_train_supgnn_data.py +++ b/0_train/2_supGNN/merge_train_FE_and_train_supgnn_data.py @@ -9,6 +9,10 @@ def merge_datasets(dataset, embedding, feats_to_drop=None): reset_ind_copy_fullds_df = copy_fullds_df.reset_index(drop=True) reset_ind_copy_emb_df = copy_emb_df.reset_index(drop=True).drop(columns=feats_to_drop) assert reset_ind_copy_fullds_df.shape[0] == reset_ind_copy_emb_df.shape[0] + + if "is_installed" in reset_ind_copy_emb_df.columns: + reset_ind_copy_emb_df.drop(columns=["is_installed"], inplace=True) + rearranged_df = reset_ind_copy_fullds_df.merge(reset_ind_copy_emb_df, on='f_0', how='left', suffixes=('_1', '_2')) assert rearranged_df.shape[0] == reset_ind_copy_fullds_df.shape[0] return rearranged_df