This repository implements an image classification pipeline based on Matrix Product States (MPS) applied to the MNIST dataset.
src/: Core Python modules for the model, preprocessing, and utilities.notebooks/: Jupyter notebooks for training, testing, and visualization.data/: Local storage for dataset files (.npy).
- Local environment: Ensure you have the required dependencies installed (NumPy, PyTorch/TensorFlow, etc.).
- Execution:
Open
notebooks/model_devl.ipynbto view the training process and classification results. The notebook imports logic directly from thesrc/directory.
Developed for research purposes in collaboration with the University of Ljubljana.