From 65388da605f9607e179520b6945efde08abd0108 Mon Sep 17 00:00:00 2001 From: olasaadi Date: Wed, 20 Jul 2022 18:39:53 +0300 Subject: [PATCH] fix docstring --- apt/utils/models/pytorch_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apt/utils/models/pytorch_model.py b/apt/utils/models/pytorch_model.py index a7f15e2..729150f 100644 --- a/apt/utils/models/pytorch_model.py +++ b/apt/utils/models/pytorch_model.py @@ -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)