prototorch_models/README.md

31 lines
675 B
Markdown
Raw Normal View History

2021-04-21 13:02:55 +00:00
# ProtoTorch Models
2021-04-21 11:13:28 +00:00
2021-04-21 11:30:50 +00:00
Pre-packaged prototype-based machine learning models using ProtoTorch and
PyTorch-Lightning.
2021-04-21 13:02:55 +00:00
## Installation
To install this plugin, simple install
[ProtoTorch](https://github.com/si-cim/prototorch) first by following the
installation instructions there and then install this plugin by doing:
```sh
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
- [X] GLVQ
- [ ] GMLVQ
- [ ] Local-Matrix GMLVQ
- [ ] Limited-Rank GMLVQ
- [ ] GTLVQ
- [ ] RSLVQ
- [ ] PLVQ
- [ ] LVQMLN
- [ ] CBC