fit_stats function

Test the fit of a model on a train and test set.