quaterion.dataset.similarity_dataset module

class SimilarityGroupDataset(dataset: Dataset)[source]

Bases: Dataset[SimilarityGroupSample]

Wrapper, which converts standard dataset of classification task into dataset, compatible with GroupSimilarityDataLoader.

Parameters:

dataset – a dataset, which return data in format: (record, label)