I am having trouble creating the submission file of my fastaiv2 model. Anyone who can help me with a code snippet of how to create the submission file using fastaiv2 kindly
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
Thankss
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
Thankss