I am new to deep learning and after watching the first video of the course, I wrote the following code for image classification for a kaggle competition (cats vs dogs).
My code is as follows:
from fastai import *
from fastai.vision import *
path = “…/input”
bs = 64
pat = re.compile(r’…/([^/…]+).\d+.jpg$’)
path_img = path + “/train/train”
fnames = get_image_files(path_img)
data = ImageDataBunch.from_name_re(path_img, fnames, pat, ds_tfms=get_transforms(),
size=224, num_workers=0, bs=bs, test=‘test1’)
learn = create_cnn(data, models.resnet34, metrics=error_rate, model_dir="/tmp/model/")
interp = ClassificationInterpretation.from_learner(learn)
losses,idxs = interp.top_losses()
It works well till here without any errors. After this, I want to make predictions on the test set on save the results in a csv file so that I can make a submission on kaggle. For this I am using the learn.get_preds() method. It however returns an output in the form of tensors. My questions are as follows:
- Am I using the right method to make predictions on the test set? If not please point me in the right direction.
- learn.get_preds() returns a list of tensors. How can I convert these tensors to class labels (1 and 0) so that I can make a submission.
- I believe learn.get_preds() works on the validation set. However, when I try to execute learn.get_preds(is_test = True) I get an error which says “unknown parameter is_test”.
How can I get around this.