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)
