torch_concepts.data.datasets.cebab.CEBaBDataset

class CEBaBDataset(pre_trained_transformer='bert-base-uncased', batch_size=32)

__init__(pre_trained_transformer='bert-base-uncased', batch_size=32)

Methods

__init__([pre_trained_transformer, batch_size])
collator()
preprocess_function(examples)