julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use squared euclidean distance in GMLVQ

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-04 14:34:00 +02:00
parent d8e017ae74
commit a1ac5a70c7
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
prototorch/models/glvq.py
View File

@ -3,7 +3,8 @@ import torch
import torchmetrics import torchmetrics
from prototorch.components import LabeledComponents from prototorch.components import LabeledComponents
from prototorch.functions.competitions import wtac from prototorch.functions.competitions import wtac
from prototorch.functions.distances import euclidean_distance from prototorch.functions.distances import (euclidean_distance,
squared_euclidean_distance)
from prototorch.functions.losses import glvq_loss from prototorch.functions.losses import glvq_loss
from prototorch.modules.prototypes import Prototypes1D from prototorch.modules.prototypes import Prototypes1D
@ -151,7 +152,7 @@ class GMLVQ(GLVQ):
protos, _ = self.proto_layer() protos, _ = self.proto_layer()
latent_x = self.omega_layer(x) latent_x = self.omega_layer(x)
latent_protos = self.omega_layer(protos) latent_protos = self.omega_layer(protos)
dis = euclidean_distance(latent_x, latent_protos) dis = squared_euclidean_distance(latent_x, latent_protos)
return dis return dis