print(df_y_ori.shape)
df_emb_ori = pd.DataFrame(
    df_y_ori,
    columns=['col_%d' % i for i in range(np.shape(df_y_ori)[1])],
    #        index=train_data['sample']
)
# df_y_ori = df_y_ori[:6180] ##use train data
## load train data
df_all = pd.read_csv("dataset/v10/df_all_new2.csv")
# print(list(train_df.columns))
# print(list(extra_test_df.columns))
df_all = df_all[df_all["split_type"] == "primary_train"]
print(df_all[["sample", "cancer_type"]])
print(df_emb_ori.shape)