mindspore.dataset.Dataset.num_classes

mindspore.dataset.Dataset.num_classes()[源代码]

获取数据集对象中所有样本的类别数目。

返回：

int，类别的数目。

样例：

>>> # dataset is an instance object of Dataset
>>> num_classes = dataset.num_classes()