Model comparison plot solved

In addition, the training and test dataset files are printed for each
splitting.