У меня есть такой код:
df_submission_gb = pd.DataFrame({'skilled': (y_pred_gb>0.5).astype(int)}, index=df_test.index)
df_submission_gb.to_csv('submission_gb_roles.csv')
df_submission_gb.tail()
t берет данные из массива y_pred_gb и помещает их в кадр df_submission_gb с индексами из df_test, но поле выходного идентификатора равно нулю:
Но у df_test есть поле идентификатора:
И это установлено, индекс на открытом:
df_test = pd.read_csv("data/test_extended", index_col="id").drop(columns=["Unnamed: 0"])