aashish24 / trame-mnist

Experimentation with AI and trame

Trame MNIST

Example application using trame for exploring MNIST dataset in the context of AI training and XAI thanks to XAITK.

• Free software: BSD License
• XAITK Saliency with MNIST
• XAI Discovery Platform | MNIST Sample Data

Installing

For the Python layer it is recommended to use conda to properly install the various ML packages.

macOS conda setup

brew install miniforge
conda init zsh

venv creation for AI

# Needed in order to get py3.9 with lzma
# PYTHON_CONFIGURE_OPTS="--enable-framework" pyenv install 3.9.9

conda create --name trame-mnist python=3.9
conda activate trame-mnist
conda install "pytorch==1.9.1" "torchvision==0.10.1" -c pytorch
conda install scipy "scikit-learn==0.24.2" "scikit-image==0.18.3" -c conda-forge

# For development
pip install -e .

# For testing
pip install trame-mnist

Run the application

conda activate trame-mnist
trame-mnist