my fork of prototorch_models
Go to file
2021-04-21 16:28:20 +02:00
examples Update mnist example 2021-04-21 16:28:20 +02:00
prototorch/models Add glvq model 2021-04-21 14:51:34 +02:00
.gitignore Update gitignore 2021-04-21 14:54:22 +02:00
LICENSE Initial commit 2021-04-21 13:13:28 +02:00
README.md Update readme 2021-04-21 15:02:55 +02:00
setup.py Update installation dependencies 2021-04-21 14:51:08 +02:00

ProtoTorch Models

Pre-packaged prototype-based machine learning models using ProtoTorch and PyTorch-Lightning.

Installation

To install this plugin, simple install ProtoTorch first by following the installation instructions there and then install this plugin by doing:

git clone https://github.com/si-cim/prototorch_models.git && cd prototorch_models
pip install -e .

The plugin should then be available for use in your Python environment as prototorch.models.

Available models

  • GLVQ
  • GMLVQ
  • Local-Matrix GMLVQ
  • Limited-Rank GMLVQ
  • GTLVQ
  • RSLVQ
  • PLVQ
  • LVQMLN
  • CBC