prediction now outputs information about confidence; it also accepts
input for the ground truth as comparsion