Not able to predict with learner in segmentation problem

[jberries]

I think you might need to create a predict augmentation function, like mentioned here (the 5th post)
So another encodes function.