prototorch_models/prototorch/models/__init__.py

"""`models` plugin for the `prototorch` package."""

from importlib.metadata import PackageNotFoundError, version

from .callbacks import (EarlyStopWithoutVal, PrototypeConvergence,
                        PruneLoserPrototypes)
from .cbc import CBC, ImageCBC
from .glvq import (
    GLVQ,
    GLVQ1,
    GLVQ21,
    GMLVQ,
    GRLVQ,
    LGMLVQ,
    LVQMLN,
    ImageGLVQ,
    ImageGMLVQ,
    SiameseGLVQ,
    SiameseGMLVQ,
)
from .lvq import LVQ1, LVQ21, MedianLVQ
from .probabilistic import CELVQ, RSLVQ, LikelihoodRatioLVQ
from .unsupervised import KNN, NeuralGas
from .vis import *

__version__ = "0.1.7"
Add Local-Matrix LVQ Also remove the use of `self.distance_fn` in favor of `self.distance_layer`. 2021-06-01 21:44:16 +00:00			"""`models` plugin for the `prototorch` package."""

Automatic Formating. 2021-04-23 15:27:47 +00:00			`from importlib.metadata import PackageNotFoundError, version`
Initial commit 2021-04-21 11:13:28 +00:00
[BUG] Early stopping does not seem to work The early stopping callback does not work as expected, and crashes at the end of max_epochs with: ``` ~/miniconda3/envs/py38/lib/python3.8/site-packages/pytorch_lightning/trainer/callback_hook.py in on_train_end(self) 155 """Called when the train ends.""" 156 for callback in self.callbacks: --> 157 callback.on_train_end(self, self.lightning_module) 158 159 def on_pretrain_routine_start(self) -> None: ~/work/repos/prototorch_models/prototorch/models/callbacks.py in on_train_end(self, trainer, pl_module) 18 def on_train_end(self, trainer, pl_module): 19 # instead, do it at the end of training loop ---> 20 self._run_early_stopping_check(trainer, pl_module) 21 22 TypeError: _run_early_stopping_check() takes 2 positional arguments but 3 were given ``` 2021-06-02 10:44:34 +00:00			`from .callbacks import (EarlyStopWithoutVal, PrototypeConvergence,`
			`PruneLoserPrototypes)`
Update models namespace 2021-05-21 11:11:59 +00:00			`from .cbc import CBC, ImageCBC`
Add Local-Matrix LVQ Also remove the use of `self.distance_fn` in favor of `self.distance_layer`. 2021-06-01 21:44:16 +00:00			`from .glvq import (`
			`GLVQ,`
			`GLVQ1,`
			`GLVQ21,`
			`GMLVQ,`
			`GRLVQ,`
			`LGMLVQ,`
			`LVQMLN,`
			`ImageGLVQ,`
			`ImageGMLVQ,`
			`SiameseGLVQ,`
			`SiameseGMLVQ,`
			`)`
Refactor non-gradient-lvq models into lvq.py 2021-05-25 18:37:34 +00:00			`from .lvq import LVQ1, LVQ21, MedianLVQ`
Add Local-Matrix LVQ Also remove the use of `self.distance_fn` in favor of `self.distance_layer`. 2021-06-01 21:44:16 +00:00			`from .probabilistic import CELVQ, RSLVQ, LikelihoodRatioLVQ`
Update Documentation Clean up project 2021-05-21 13:42:45 +00:00			`from .unsupervised import KNN, NeuralGas`
Add models to the prototorch.models namespace 2021-05-07 13:23:52 +00:00			`from .vis import *`

Bump version: 0.1.6 → 0.1.7 2021-05-11 15:18:29 +00:00			`__version__ = "0.1.7"`