integrate reviews from ChristophRaab:master

This commit is contained in:
Alexander Engelsberger 2021-05-27 09:43:02 +02:00
parent 00615ae837
commit c204bc8e1f

View File

@ -1,12 +1,9 @@
import torch
from prototorch.functions.distances import (euclidean_distance_matrix,
tangent_distance)
from prototorch.functions.helper import _check_shapes, _int_and_mixed_shape
from prototorch.functions.distances import euclidean_distance_matrix
from prototorch.functions.normalization import orthogonalization
from prototorch.modules.prototypes import Prototypes1D
from torch import nn
class GTLVQ(nn.Module):
r""" Generalized Tangent Learning Vector Quantization