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

score

Browse source
This commit is contained in:
olasaadi 2022-05-19 04:40:02 +03:00
parent 7539ca0ead
commit e0385b0d04
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -207,8 +207,10 @@ class PyTorchClassifier(PyTorchModel):
def score(self, test_data: Dataset, **kwargs):
"""
Score the model using test data.
:param test_data: Test data.
:type train_data: `Dataset`
:return: the score as float (between 0 and 1)
"""
pass
y = check_and_transform_label_format(test_data.get_labels(), self._art_model.nb_classes)
predicted = self.predict(test_data)
return np.count_nonzero(np.argmax(y, axis=1) == np.argmax(predicted, axis=1)) / predicted.shape[0]