Re-name class to state testing. Add a class for split training and whole
dataset training. Ability to save RMSE and MSE