I seem to get more stable results when I do
Code:
if (Train) {
  Hedge = 2;
} else {
  Hedge = 4;
  set(NFA);
}