Support returning of raw predictions via socket