def submit(ypred): from dataloader import DataLoader dataloader = DataLoader(config) df_test = dataloader.load_test_data() id_test = df_test['id'] pd.DataFrame({"id": id_test, "relevance": ypred}).to_csv('submission.csv',index=False)