Test set for multi label classification model?

I have been successful at training a model for multilabel classification.
Yet, I couldn’t add a testing set to check its performance on a set of new images:

To train the model with cross-validation, I use do the following:

for train_index, val_index in skf.split(df.index, df['Tags']):
  src = (ImageList.from_df(df, '/content/gdrive/My Drive/Dataset/500 dataset together')
       .label_from_df(label_delim=' '))
  data_fold = (src.transform(None, size=224).databunch().normalize(imagenet_stats))
  learn = cnn_learner(data_fold, models.resnet18, metrics=acc_02)
  learn.fit_one_cycle(7, slice(lr))
  loss,acc = learn.validate()

Can you please help? Thanks in advance.

Can you tell us what error you are getting?
In docs.fast.ai there is a multi-label classification example .

You can try something like below.

learn.data.add_test(ImageList.from_df(test_df, PATH, folder='test'))
preds,_ = learn.get_preds(ds_type=DatasetType.Test)

Hope this helps.

1 Like

Dear @PoonamV, Thank you for your quick response. But don’t you think I need to apply all things that were applied for the test data, such as normalization and the corresponding labels to compute test accuracy?

normalization will be applied automatically. If you need to apply transformations to test the same as train data. You need to use learn.TTA().

If you have labels, you need to treat it as validation set and add it to Databunch with its labels and compute accuracy. You need to pass as DatasetType.Valid.

1 Like

You can also create a new databunch and then use learn.validate on that databunch. For this databunch use split_none() for no validation split.

1 Like