We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b970b30 commit 84b8c55Copy full SHA for 84b8c55
art/utils.py
@@ -670,7 +670,7 @@ def floats_to_one_hot(labels: np.ndarray):
670
671
672
def check_and_transform_label_format(
673
- labels: np.ndarray, nb_classes: Optional[int] = None, return_one_hot: bool = True
+ labels: np.ndarray, nb_classes: Optional[int], return_one_hot: bool = True
674
) -> np.ndarray:
675
"""
676
Check label format and transform to one-hot-encoded labels if necessary
0 commit comments