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

enhance calculation of nb classes + tests (#45)

Browse source 
* update get_nb_classes method to handle 1-hot and scalar input
This commit is contained in:
Shlomit Shachor 2022-07-05 11:32:17 +03:00 • committed by GitHub
parent 50317a8d67
commit e25e58b253
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 62 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

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

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