xailib.models.keras_classifier_wrapper

Keras classifier wrapper for XAI-Lib.

This module provides a wrapper class for Keras/TensorFlow classifiers, allowing them to be used with XAI-Lib explainers.

Classes:

keras_classifier_wrapper: Wrapper for Keras classifiers.

Example

>>> from tensorflow import keras
>>> from xailib.models.keras_classifier_wrapper import keras_classifier_wrapper
>>>
>>> # Build and train your Keras model
>>> model = keras.Sequential([...])
>>> model.fit(X_train, y_train)
>>>
>>> # Wrap it for use with XAI-Lib
>>> bb = keras_classifier_wrapper(model)
>>>
>>> # Now use with any explainer
>>> from xailib.explainers.gradcam_explainer import GradCAMImageExplainer
>>> explainer = GradCAMImageExplainer(bb)

Classes

keras_classifier_wrapper(classifier)

Wrapper class for Keras/TensorFlow classifiers.