def distmats_and_labels() -> List[Tuple[Tensor, List[int]]]: """ Returns: list of distance matrices and valid labels """ num_batches = 100 labels_pk = generate_valid_labels(num=num_batches) labels_list, _, _ = zip(*labels_pk) distmats = [] for labels in labels_list: n = len(labels) distmats.append(tensor(squareform(torch.rand(int(n * (n - 1) / 2))))) return list(zip(distmats, labels_list))
def features_and_labels() -> List[Tuple[Tensor, List[int]]]: """ Returns: list of features and valid labels """ num_batches = 100 features_dim = 10 labels_pk = generate_valid_labels(num=num_batches) labels_list, _, _ = zip(*labels_pk) features = [] for labels in labels_list: features.append(torch.rand(size=(len(labels), features_dim))) return list(zip(features, labels_list))