StochasticSupervisor::trainFor() improvements.