A method and system for assessing a machine learning model providing a prediction as to the disease state of a patient from a 2D or 3D image of the patient or a sample obtained therefrom. The machine learning model produces a prediction of the disease state from the image. The method involves presenting on a display of a workstation the image of the patient or a sample obtained therefrom along with a risk score or classification associated with the prediction. The image is further augmented with highlighting to indicate one or more regions in the image which affected the prediction produced by the machine learning model. Tools are provided by which the user may highlight one or more regions of the image which the user deems to be suspicious for the disease state. Inference is performed on the user-highlighted areas by the machine learning model. The results of the inference are presented to the user via the display.