apunkt/ai-privacy-toolkit
1
0
Fork 0
mirror of https://github.com/IBM/ai-privacy-toolkit.git synced 2026-07-11 16:22:10 +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
tests/test_model.py
View file

@ -179,7 +179,7 @@ def test_blackbox_classifier_predictions_multi_label_cat():
def test_blackbox_classifier_predictions_multi_label_binary():
(x_train, y_train), (x_test, y_test) = dataset_utils.get_iris_dataset_np()
# make multi-label categorical
# make multi-label binary
y_train = np.column_stack((y_train, y_train, y_train))
y_train[y_train > 1] = 1
pred_train = y_train.copy().astype(float)