Labeled and unlabeled data python