Comparison on train and test datasets