def get_data_splits(): validation_dataset, split_train_dataset = split_dataset(train_dataset, prop.VAL_SIZE) return split_train_dataset, validation_dataset, test_dataset
def get_policy_training_splits(): test_dataset, train_dataset = split_dataset(full_dataset, prop.POLICY_TEST_SIZE) validation_dataset, split_train_dataset = split_dataset(train_dataset, prop.VAL_SIZE) return split_train_dataset, validation_dataset, test_dataset
def get_data_splits(): validation_dataset, split_train_dataset = split_dataset(train_dataset, prop.VAL_SIZE) # train_size = 2000 # split_train_dataset, _ = split_dataset(split_train_dataset, train_size) return split_train_dataset, validation_dataset, test_dataset