changes to n_samples and bellman basis training structure