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))
