quaterion.dataset.similarity_dataset 模块

class SimilarityGroupDataset(dataset: Dataset)

基类：Dataset[SimilarityGroupSample]

包装器，将标准的分类任务数据集转换为与 GroupSimilarityDataLoader 兼容的数据集。

参数：

dataset – 一个数据集，其返回数据的格式为：(record, label)