Given all predictions and all true labels, plot histograms of top-k most frequently predicted classes for each true class. Args: writer (SummaryWriter object): a tensorboard SummaryWriter object. cmtx (ndarray): confusion matrix. num_classes (int): total number
(
writer,
cmtx,
num_classes,
k=10,
global_step=None,
subset_ids=None,
class_names=None,
figsize=None,
)
| 277 | |
| 278 | |
| 279 | def plot_hist( |
| 280 | writer, |
| 281 | cmtx, |
| 282 | num_classes, |
| 283 | k=10, |
| 284 | global_step=None, |
| 285 | subset_ids=None, |
| 286 | class_names=None, |
| 287 | figsize=None, |
| 288 | ): |
| 289 | """ |
| 290 | Given all predictions and all true labels, plot histograms of top-k most |
| 291 | frequently predicted classes for each true class. |
| 292 | |
| 293 | Args: |
| 294 | writer (SummaryWriter object): a tensorboard SummaryWriter object. |
| 295 | cmtx (ndarray): confusion matrix. |
| 296 | num_classes (int): total number of classes. |
| 297 | k (int): top k to plot histograms. |
| 298 | global_step (Optional[int]): current step. |
| 299 | subset_ids (list of ints, optional): class indices to plot histogram. |
| 300 | mapping (list of strings): names of all classes. |
| 301 | figsize (Optional[float, float]): the figure size of the confusion matrix. |
| 302 | If None, default to [6.4, 4.8]. |
| 303 | """ |
| 304 | if subset_ids is None or len(subset_ids) != 0: |
| 305 | if subset_ids is None: |
| 306 | subset_ids = set(range(num_classes)) |
| 307 | else: |
| 308 | subset_ids = set(subset_ids) |
| 309 | # If class names are not provided, use their indices as names. |
| 310 | if class_names is None: |
| 311 | class_names = list(range(num_classes)) |
| 312 | |
| 313 | for i in subset_ids: |
| 314 | pred = cmtx[i] |
| 315 | hist = vis_utils.plot_topk_histogram( |
| 316 | class_names[i], |
| 317 | torch.Tensor(pred), |
| 318 | k, |
| 319 | class_names, |
| 320 | figsize=figsize, |
| 321 | ) |
| 322 | writer.add_figure( |
| 323 | tag="Top {} predictions by classes/{}".format( |
| 324 | k, class_names[i] |
| 325 | ), |
| 326 | figure=hist, |
| 327 | global_step=global_step, |
| 328 | ) |
| 329 | |
| 330 | |
| 331 | def add_ndim_array( |