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

Add support for xgboost XGBClassifier (#53)

Browse source
This commit is contained in:
abigailt 2022-07-28 16:31:08 +03:00 • committed by abigailgold
parent a13415ad67
commit a9e2a35e18
3 changed files with 106 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
apt/utils/models/__init__.py
View file

@ -3,3 +3,4 @@ from apt.utils.models.model import Model, BlackboxClassifier, ModelOutputType, S
check_correct_model_output
from apt.utils.models.sklearn_model import SklearnModel, SklearnClassifier, SklearnRegressor
from apt.utils.models.keras_model import KerasClassifier, KerasRegressor
from apt.utils.models.xgboost_model import XGBoostClassifier