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

apply changes after rebase with wrappers

Browse source
This commit is contained in:
olasaadi 2022-03-10 13:49:05 +02:00
parent 6afb175d6f
commit b4eddabd37
4 changed files with 32 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

4
apt/anonymization/anonymizer.py
View file

@ -5,7 +5,7 @@ from collections import Counter
from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
from sklearn.preprocessing import OneHotEncoder
from apt.utils.datasets import BaseDataset, Data
from apt.utils.datasets import ArrayDataset
from typing import Union, Optional
@ -38,7 +38,7 @@ class Anonymize:
self.categorical_features = categorical_features
self.is_regression = is_regression
def anonymize(self, dataset: BaseDataset) -> Union[np.ndarray, pd.DataFrame]:
def anonymize(self, dataset: ArrayDataset) -> Union[np.ndarray, pd.DataFrame]:
"""
Method for performing model-guided anonymization.