Skip to content

Commit 9ff3d28

Browse files
committed
fix img_filename column
1 parent 2f00241 commit 9ff3d28

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/analysing-image-classification-dataset.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -426,10 +426,10 @@
426426
"df_annot = df_annot[['path', 'noisy_labels_0']]\n",
427427
"\n",
428428
"# rename columns to fastdup's column names\n",
429-
"df_annot = df_annot.rename({'noisy_labels_0': 'label', 'path': 'img_filename'}, axis='columns')\n",
429+
"df_annot = df_annot.rename({'noisy_labels_0': 'label', 'path': 'filename'}, axis='columns')\n",
430430
"\n",
431431
"# create split column\n",
432-
"df_annot['split'] = df_annot['img_filename'].apply(lambda x: x.split(\"/\")[0])\n",
432+
"df_annot['split'] = df_annot['filename'].apply(lambda x: x.split(\"/\")[0])\n",
433433
"\n",
434434
"# map label ids to regular labels\n",
435435
"df_annot['label'] = df_annot['label'].map(label_map)\n",

0 commit comments

Comments
 (0)