Solved problem with load function of train_holdout_validation class