the Pytorch framework was used to visualize the salient features that contributed the most to the prediction output