Improve verbose output during training; improve on_epoch_finished interface.
Make tensor type of X variable; allows for training with 2d data.