This is the v1 snippet from the starter notebook:

import os

submission = pd.DataFrame()

submission["ID"] = test_df["img_IDS"]

for i, c in enumerate(learn.data.classes):

print(c)

submission[c] = preds_test[:,i]

submission.head()

Replace learn.data.classes with learn.data.vocab