trainig_test_comparison function

Compare performance of model between training and test set