torch_concepts.data.utils.assign_random_values

assign_random_values(concept, random_prob=[0.5, 0.5], values=[0, 1])

Create a vector of random values for each sample in concepts. :param concepts: Tensor of shape (N) containing concept values (e.g. digit labels 0-9). :param random_prob: List of probabilities for each value. :param values: List of output values corresponding to each probability.

Returns:

Tensor of shape (N) containing final values.

Return type:

outputs