apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-06-29 15:59:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Externalize common test code to methods.

Browse source 
Support for sparse matrix.

Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2023-08-10 12:57:59 +03:00
parent 3de93a87f1
commit f85fc87bdd
3 changed files with 122 additions and 323 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -46,7 +46,7 @@ class SklearnClassifier(SklearnModel):
def __init__(self, model: BaseEstimator, output_type: ModelOutputType, black_box_access: Optional[bool] = True,
unlimited_queries: Optional[bool] = True, **kwargs):
super().__init__(model, output_type, black_box_access, unlimited_queries, **kwargs)
self._art_model = ArtSklearnClassifier(model)
self._art_model = ArtSklearnClassifier(model, preprocessing=None)
def fit(self, train_data: Dataset, **kwargs) -> None:
"""