dcase_models.util.tag_probabilities_to_tag_list¶
-
dcase_models.util.tag_probabilities_to_tag_list(tag_probabilities, label_list, threshold=0.5)[source]¶ Convert a tag probabilites matrix to a tag list.
Parameters: - tag_probabilities : ndarray
Shape (N_times, N_classes)
- label_list : list of str
Label list
- threshold : float
Threshold to decide if a tag is present.
Returns: - list
List of tags. e.g. [‘dog’, ‘cat’, …]