torch_concepts.data.utils.assign_values_based_on_intervals

assign_values_based_on_intervals(concept, intervals, values)

Create a vector of values (0 or 1) for each sample in concepts based on intervals given. If a concept value belongs to interval[i], it gets an output value randomly chosen among values[i]. :param concept: Tensor of shape (N) containing concept values (e.g. digit labels 0-9). :param intervals: List of lists, each inner list contains the values defining an interval. :param values: List of lists of output values corresponding to each interval.

Returns:

Tensor of shape (N) containing final values.

Return type:

outputs