From c0dbb200e77e6528915fefdd94f2c9d941c4dd47 Mon Sep 17 00:00:00 2001 From: olasaadi Date: Thu, 10 Mar 2022 13:04:39 +0200 Subject: [PATCH] add dataset wrapper to docstring --- apt/anonymization/anonymizer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/apt/anonymization/anonymizer.py b/apt/anonymization/anonymizer.py index 0f90a2d..d7452b1 100644 --- a/apt/anonymization/anonymizer.py +++ b/apt/anonymization/anonymizer.py @@ -42,9 +42,9 @@ class Anonymize: """ Method for performing model-guided anonymization. - :param x: The training data for the model. If provided as a pandas dataframe, may contain both numeric and - categorical data. - :param y: The predictions of the original model on the training data. + :param dataset: Data wrapper Containing The training data for the model and ehe predictions of the + original model on the training data. If provided as a pandas + dataframe, may contain both numeric and categorical data. :return: An array containing the anonymized training dataset. """ if type(dataset.x) == np.ndarray: