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

Formatting (#68)

Browse source 
Fix most flake/lint errors and ignore a few others

Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailgold 2022-12-25 15:13:57 +02:00 • committed by GitHub
parent b47ba24906
commit d52fcd0041
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
16 changed files with 91 additions and 92 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -38,7 +38,7 @@ class XGBoostClassifier(XGBoostModel):
:type unlimited_queries: boolean, optional
"""
def __init__(self, model: XGBClassifier, output_type: ModelOutputType, input_shape: Tuple[int, ...],
nb_classes: int,black_box_access: Optional[bool] = True,
nb_classes: int, 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 = ArtXGBoostClassifier(model, nb_features=input_shape[0], nb_classes=nb_classes)