Learned mixture model for planning in stochastic environments