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

fix bug and update test_model

Browse source
This commit is contained in:
olasaadi 2022-03-10 12:56:41 +02:00
parent fd9f134336
commit 300e391432
3 changed files with 19 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

5
apt/utils/models/sklearn_model.py
View file

@ -46,7 +46,10 @@ class SklearnClassifier(SklearnModel):
:type y: `np.ndarray` or `pandas.DataFrame`
"""
encoder = OneHotEncoder(sparse=False)
y_encoded = encoder.fit_transform(y.reshape(-1, 1))
if type(y) == np.ndarray:
y_encoded = encoder.fit_transform(y.reshape(-1, 1))
else:
y_encoded = encoder.fit_transform(y.values.reshape(-1, 1))
self._art_model.fit(x, y_encoded, **kwargs)
def predict(self, x: np.ndarray, **kwargs) -> np.ndarray: