Skip to main content

AI Engineering Interpretability Utils

Helper modules for AI Engineering Interpretability Bootcamp implementations.

This package provides reusable utilities for:

  • Loading and preprocessing common datasets (Adult, Diabetes, Bank Marketing, Gas Turbine)
  • Model training and evaluation helpers
  • Visualization utilities for interpretability

Installation

pip install aieng-interp-utils

Usage

from aieng.interp.data import load_adult_dataset, preprocess_adult_data
from aieng.interp.models import train_pytorch_model, get_device
from aieng.interp.visualization import plot_confusion_matrix, plot_roc_curve

# Load and preprocess data
df = load_adult_dataset("path/to/data")
X_train, X_test, y_train, y_test = preprocess_adult_data(df)

# Train model
device = get_device()
model = train_pytorch_model(model, X_train, y_train, device=device)

# Visualize results
plot_roc_curve(y_test, y_pred_proba)
plot_confusion_matrix(y_test, y_pred)

License

MIT

Metadata

Release files for aieng-interp-utils 1.0.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for aieng-interp-utils 1.0.0
File Size Uploaded Provenance
aieng_interp_utils-1.0.0.tar.gz 7.1 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for aieng-interp-utils 1.0.0
File Interpreter ABI Platform
aieng_interp_utils-1.0.0-py3-none-any.whl Python 3 none any Details

Total release size: 17.1 kB

Release files / aieng_interp_utils-1.0.0.tar.gz

Download URL aieng_interp_utils-1.0.0.tar.gz
Size 7.1 kB
Tags Source
SHA-256 checksum
How to use checksums
f98e5771561eab734bb84325f927c3d83ee1c34f352fcf91af9dba8e8d25e216
BLAKE2b-256 checksum
How to use checksums
306fe511af4f03f03341e403f200f3fec19fe32d3bdff170fe1709aa16dd0b51
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.13.7

Release files / aieng_interp_utils-1.0.0-py3-none-any.whl

Download URL aieng_interp_utils-1.0.0-py3-none-any.whl
Size 9.9 kB
Tags Python 3
SHA-256 checksum
How to use checksums
bf32bcbe4fee911fcf2ac1ed56dc1c0dd14ad29d0d327c1075c83e524ab28284
BLAKE2b-256 checksum
How to use checksums
e2d1d26e880fdaba41a2fbe7d54be0ed184fd9d1ddf78266b79ed360a54c3c2a
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.13.7

Security for aieng-interp-utils 1.0.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

1.0.0 This release

2 release files No provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page