Skip to main content

This package provides means of explainable classification of images with LLM-generated text descriptions, that are then analyzed using the Self Model Entities Related (SMER) explanation technique.

Project description

SMERVisual

SMERVisual is a Python package designed for explainable machine learning using the SMER method. It provides an intuitive interface for analyzing image classification models by highlighting key features and generating bounding boxes around important words in source images.

The package supports both OpenAI API models and local language models, offering flexibility in model selection.


Installation

Install SMERVisual using pip:

pip install smer-visual

Features

ImageClassifier

The ImageClassifier class performs logistic regression-based classification on image datasets while computing LIME and SMER values for explainability. Key features include:

  • Classification of image datasets using pre-trained language models
  • Computation of LIME and SMER scores for each word in the dataset
  • Visualization of most important words for explainability

BoundingBoxGenerator

The BoundingBoxGenerator class enhances model interpretability by overlaying bounding boxes on images, highlighting critical words identified in classification. It:

  • Uses ImageClassifier output to detect significant words
  • Supports vision models from OpenAI and open-source alternatives
  • Enables zero-shot object detection for explainable AI

Usage Example

Here's a simple way to use SMERVisual for explainable image classification:

from smer_visual import ImageClassifier, BoundingBoxGenerator

# Initialize and train the classifier
classifier = ImageClassifier(openai_model="openai", openai_key='123abc123')
classifier(data="path/to/dataset")
influential_words = classifier.get_top_words()

# Generate bounding boxes for top words
bbox_generator = BoundingBoxGenerator(data='path/to/dataset', top_words=influential_words,
                                      local_model_path='path/to/local/model')
bbox_generator()

Why Use SMERVisual?

  • Explainable AI – Provides insight into model decision-making
  • Model-Agnostic – Compatible with OpenAI APIs and open-source models
  • Zero-Shot Detection – No additional training data required
  • Easy Integration – Simple API for seamless use with existing machine learning workflows

Contributing

Contributions are welcome. If you’d like to contribute, follow these steps:

  1. Fork the repository
  2. Create a new branch (git checkout -b feature-branch)
  3. Commit changes (git commit -m "Add new feature")
  4. Push to the branch (git push origin feature-branch)
  5. Open a pull request

License

SMERVisual is released under the MIT License.


Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

smer_visual-1.0.1.tar.gz (13.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

smer_visual-1.0.1-py3-none-any.whl (13.1 kB view details)

Uploaded Python 3

File details

Details for the file smer_visual-1.0.1.tar.gz.

File metadata

  • Download URL: smer_visual-1.0.1.tar.gz
  • Upload date:
  • Size: 13.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.10.0

File hashes

Hashes for smer_visual-1.0.1.tar.gz
Algorithm Hash digest
SHA256 23b30fb7bce34f21892cafb21b586b196e859ada46c2dda19526620d76e6fec4
MD5 bffada9d3ca9d8c82ba164297dbab2bb
BLAKE2b-256 54ef0da02565b5399436d279b2e185fd5d661086d6beb0292cd170e042f907d3

See more details on using hashes here.

File details

Details for the file smer_visual-1.0.1-py3-none-any.whl.

File metadata

  • Download URL: smer_visual-1.0.1-py3-none-any.whl
  • Upload date:
  • Size: 13.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.10.0

File hashes

Hashes for smer_visual-1.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 af5fd49653eb2c8ef10d24e76661ec339573297fa695f3f3ed8710cfc5e4ba92
MD5 57b395f4b76ebe52f8c088c1a70c328e
BLAKE2b-256 559f5d1135d9bcaa69641bca914a02d58e81793abc61e245866d5df4cefe6c66

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page