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

fix for keras model check_and_transform_label_format requires nb_classes

Browse source
This commit is contained in:
natali 2022-07-02 23:29:49 +03:00 • committed by abigailgold
parent c6eb553a9f
commit 50317a8d67
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -57,7 +57,7 @@ class KerasClassifier(KerasModel):
:type train_data: `Dataset`
:return: None
"""
y_encoded = check_and_transform_label_format(train_data.get_labels())
y_encoded = check_and_transform_label_format(train_data.get_labels(), self._art_model.nb_classes)
self._art_model.fit(train_data.get_samples(), y_encoded, **kwargs)
def predict(self, x: Dataset, **kwargs) -> OUTPUT_DATA_ARRAY_TYPE: