incerto.active.split_labeled_unlabeled

incerto.active.split_labeled_unlabeled(data, labels=None, labeled_indices=None, unlabeled_indices=None)

Split data into labeled and unlabeled sets.

Parameters:

• data (Tensor) – Full dataset (N, ...)

• labels (Optional[Tensor]) – Labels for all data (N,), can have -1 for unlabeled

• labeled_indices (Optional[Tensor]) – Indices of labeled samples

• unlabeled_indices (Optional[Tensor]) – Indices of unlabeled samples

Return type:

Tuple[Tensor, Tensor, Optional[Tensor], Optional[Tensor]]

Returns:

Tuple of (x_labeled, x_unlabeled, y_labeled, labeled_indices)