| import requests | |
| import numpy as np | |
| from typing import Dict | |
| def create_labels_legend(concept_scores: np.ndarray, | |
| labels: Dict[int, str], | |
| top_k=2): | |
| concept_categories = np.argsort(concept_scores, axis=1)[:, ::-1][:, :top_k] | |
| concept_labels_topk = [] | |
| for concept_index in range(concept_categories.shape[0]): | |
| categories = concept_categories[concept_index, :] | |
| concept_labels = [] | |
| for category in categories: | |
| score = concept_scores[concept_index, category] | |
| label = f"{labels[category].split(',')[0]}:{score:.2f}" | |
| concept_labels.append(label) | |
| concept_labels_topk.append("\n".join(concept_labels)) | |
| return concept_labels_topk | |