Error when running retrain.py

[atapley]

There seems to be an error in retrain.py where the output of _train() does not match what is expected in train() causing an error for too many variables to unpack (line 364).

Not sure if this error is due to a change in the _train method that was not reflected in train or if there is something missing somewhere.