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

Keras regression support

Browse source
This commit is contained in:
abigailt 2022-07-20 14:17:25 +03:00 • committed by abigailgold
parent a7d156660e
commit 77a6e08c8e
4 changed files with 85 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2,4 +2,4 @@ from apt.utils.models.model import Model, BlackboxClassifier, ModelOutputType, S
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
from apt.utils.models.keras_model import KerasClassifier, KerasRegressor