test: add unit tests

2022-03-30 15:12:33 +02:00
parent 9da47b1dba
commit 7d3f59e54b
6 changed files with 218 additions and 26 deletions
--- a/prototorch/models/init.py
+++ b/prototorch/models/init.py
@@ -26,13 +26,11 @@ from .lvq import (
 )
 from .probabilistic import (
    CELVQ,
-    PLVQ,
    RSLVQ,
    SLVQ,
 )
 from .unsupervised import (
    GrowingNeuralGas,
-    HeskesSOM,
    KohonenSOM,
    NeuralGas,
 )