jax_privacy.batch_selection.PartitionType

class jax_privacy.batch_selection.PartitionType(value)

Bases: Enum

An enum specifying how examples should be assigned to groups.

Attributes

INDEPENDENT	Each example will be assigned to a group independently at random.
EQUAL_SPLIT	Examples will be shuffled and then split into groups of equal size.

INDEPENDENT = 1

Each example will be assigned to a group independently at random.

EQUAL_SPLIT = 2

Examples will be shuffled and then split into groups of equal size.