Added train, get predictions functions to Validator.
Added test function to Validator.
Added visualization of predicted labels.