fixing print of ytrain.shape