Simpler intrvfl setup

hyperdimensional-computing · Mar 6, 2024 · fc3b80c · fc3b80c
1 parent 19ca163
commit fc3b80c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchhd/tests/test_models.py b/torchhd/tests/test_models.py
@@ -113,7 +113,9 @@ def test_initialization(self, dtype):
         assert model.weight.device.type == device.type
 
     def test_fit_ridge_regression(self):
-        samples = torch.eye(10, 12)
+        a = torch.randn(10)
+        b = torch.randn(12)
+        samples = torch.outer(a, b)
         targets = torch.arange(10)
 
         model = models.IntRVFL(12, 1245, 10)