concepts.benchmark.logic_induction.family.FamilyTreeDataset#
- class FamilyTreeDataset[source]#
Bases:
Dataset
Methods
Attributes
- __add__(other)#
- Parameters:
other (Dataset[T_co])
- Return type:
ConcatDataset[T_co]
- __init__(nr_people, epoch_size, task, p_marriage=0.8, balance_sample=False, np_random=None, seed=None)[source]#
- Parameters:
np_random (RandomState | None)
seed (int | None)
- __new__(**kwargs)#
- available_tasks = ['has-father', 'has-daughter', 'has-sister', 'parents', 'grandparents', 'uncle', 'maternal-great-uncle']#