prototorch_models/README.md
Jensun Ravichandran ee39ac516d Update readme
2021-04-21 15:02:55 +02:00

675 B

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