xailib.models.sklearn_classifier_wrapper

Scikit-learn classifier wrapper for XAI-Lib.

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

Classes:

sklearn_classifier_wrapper: Wrapper for scikit-learn classifiers.

Example

>>> from sklearn.ensemble import RandomForestClassifier
>>> from xailib.models.sklearn_classifier_wrapper import sklearn_classifier_wrapper
>>>
>>> # Train your model
>>> rf = RandomForestClassifier()
>>> rf.fit(X_train, y_train)
>>>
>>> # Wrap it for use with XAI-Lib
>>> bb = sklearn_classifier_wrapper(rf)
>>>
>>> # Now use with any explainer
>>> from xailib.explainers.lime_explainer import LimeXAITabularExplainer
>>> explainer = LimeXAITabularExplainer(bb)

Classes

sklearn_classifier_wrapper(classifier)

Wrapper class for scikit-learn classifiers.