Use iterators for learnModel and predict
Partial write to use IterativeSofImpute