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

Initial support+test for pytorch multi-label binary classifier

Browse source 
Signed-off-by: abigailt <abigailt@il.ibm.com>
This commit is contained in:
abigailt 2024-02-19 14:16:03 +02:00
parent f197199e54
commit 79534b69db
4 changed files with 193 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,6 @@
from apt.utils.models.model import Model, BlackboxClassifier, ModelOutputType, ScoringMethod, \
BlackboxClassifierPredictions, BlackboxClassifierPredictFunction, get_nb_classes, is_one_hot, \
check_correct_model_output
check_correct_model_output, is_multi_label, is_multi_label_binary
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