From ed03ab168e2d8c0b11e5196a04dc34d466d3e991 Mon Sep 17 00:00:00 2001
From: Jensun Ravichandran <jjensun@gmail.com>
Date: Mon, 10 May 2021 14:09:25 +0200
Subject: [PATCH] [BUGFIX] Fix lambda_matrix property in GMLVQ

---
 prototorch/models/glvq.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/prototorch/models/glvq.py b/prototorch/models/glvq.py
index 308bfae..bd8c3be 100644
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@@ -200,8 +200,8 @@ class GMLVQ(GLVQ):
 
     @property
     def lambda_matrix(self):
-        omega = self.omega_layer.weight
-        lam = omega @ omega.T
+        omega = self.omega_layer.weight  # (latent_dim, input_dim)
+        lam = omega.T @ omega
         return lam.detach().cpu()
 
     def show_lambda(self):