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
