debug: Add prints to find out where predictions fail

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>