ai-privacy-toolkit/apt/utils/models
Shlomit Shachor e25e58b253
enhance calculation of nb classes + tests (#45)
* update get_nb_classes method to handle 1-hot and scalar input
2022-07-05 11:32:17 +03:00
..
__init__.py enhance calculation of nb classes + tests (#45) 2022-07-05 11:32:17 +03:00
keras_model.py fix for keras model check_and_transform_label_format requires nb_classes 2022-07-03 09:59:41 +03:00
model.py enhance calculation of nb classes + tests (#45) 2022-07-05 11:32:17 +03:00
sklearn_model.py enhance calculation of nb classes + tests (#45) 2022-07-05 11:32:17 +03:00