pytorchTutorial
pytorchTutorial copied to clipboard
Update 16_tensorboard.py
Based on pytorch docs, the parameter "labels" of add_pr_curve() should be ground truth data. But in Line 141, class_labels append our own predictions. Therefore, I substitute "labels" for "predicted".