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

Check for mismatch between model output type and actual output

Browse source
This commit is contained in:
abigailt 2022-07-19 08:43:19 +03:00 • committed by abigailgold
parent bc7ab0cc7f
commit 1cc73b3da1
5 changed files with 75 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,4 +1,5 @@
from apt.utils.models.model import Model, BlackboxClassifier, ModelOutputType, ScoringMethod, \
BlackboxClassifierPredictions, BlackboxClassifierPredictFunction, get_nb_classes, is_one_hot
BlackboxClassifierPredictions, BlackboxClassifierPredictFunction, get_nb_classes, is_one_hot, \
check_correct_model_output
from apt.utils.models.sklearn_model import SklearnModel, SklearnClassifier, SklearnRegressor
from apt.utils.models.keras_model import KerasClassifier