Change diagram code to look at correct predictions and a small fix
Change diagram code to look at correct predictions and a small fix

jakelever/knowledgediscovery,jakelever/knowledgediscovery,jakelever/knowledgediscovery,jakelever/knowledgediscovery