apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-04-24 20:36:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add dataset wrapper to docstring

Browse source
This commit is contained in:
olasaadi 2022-03-10 13:04:39 +02:00
parent 300e391432
commit c0dbb200e7
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
apt/anonymization/anonymizer.py
View file

@ -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: