I think I find a solution here. So far I split the dataset into half training and half test in the dataloader, and the CV is only trained, tested and evaluated on the training part of the dataset. I use the other half to plot the test result. I am currently plotting it with my own plotting script. Please let me know if you have any comments.