Code snippet for medium article

plantvillage_train.py

learn = cnn_learner(dls, resnet34, pretrained=True, metrics=accuracy).to_fp16()

learn.fit_one_cycle(4)
learn.save('stage_1')

#Understand which classes are misclassified
interp = ClassificationInterpretation.from_learner(learn)
losses,idxs = interp.top_losses()
interp.plot_top_losses(9, figsize=(15,10))
interp.most_confused(min_val=3)

#Create confusion matrix
interp.plot_confusion_matrix(figsize=(12,12), dpi=60)

#Unfreeze and retrain
learn.load('stage_1')
learn.unfreeze()

#Inference
learn.show_results(figsize=(15,15))
learn.lr_find()