apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-06-08 15:05:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix docstring

Browse source
This commit is contained in:
olasaadi 2022-07-20 18:39:53 +03:00
parent c2c7a01078
commit 65388da605
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
apt/utils/models/pytorch_model.py
View file

@ -302,7 +302,7 @@ class PyTorchClassifier(PyTorchModel):
"""
Score the model using test data.
:param test_data: Test data.
:type train_data: `Dataset`
:type test_data: `Dataset`
:return: the score as float (between 0 and 1)
"""
y = check_and_transform_label_format(test_data.get_labels(), self._art_model.nb_classes)