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

predict_latent no longer returns numpy

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-15 12:52:16 +02:00
parent ebc42a4aa8
commit b7684ae512
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
prototorch/models/glvq.py
View File

@ -171,7 +171,7 @@ class SiameseGLVQ(GLVQ):
latent_protos = self.backbone_dependent(protos)
d = euclidean_distance(x, latent_protos)
y_pred = wtac(d, plabels)
return y_pred.numpy()
return y_pred
class GRLVQ(GLVQ):
@ -206,7 +206,7 @@ class GRLVQ(GLVQ):
latent_protos = protos @ torch.diag(self.relevances)
d = squared_euclidean_distance(x, latent_protos)
y_pred = wtac(d, plabels)
return y_pred.numpy()
return y_pred
class GMLVQ(GLVQ):
@ -261,7 +261,7 @@ class GMLVQ(GLVQ):
latent_protos = self.omega_layer(protos)
d = squared_euclidean_distance(x, latent_protos)
y_pred = wtac(d, plabels)
return y_pred.numpy()
return y_pred
class ImageGMLVQ(GMLVQ, PrototypeImageModel):
@ -306,4 +306,4 @@ class LVQMLN(GLVQ):
latent_protos, plabels = self.proto_layer()
d = euclidean_distance(x, latent_protos)
y_pred = wtac(d, plabels)
return y_pred.numpy()
return y_pred