add variation of predict that also returns the confidence