[HOTFIX] Add missing iris.py and fix knnc bug

This commit is contained in:
Jensun Ravichandran
2021-05-11 17:20:48 +02:00
parent 09256956f3
commit b6145223c8
3 changed files with 19 additions and 4 deletions

View File

@@ -4,7 +4,6 @@ import unittest
import numpy as np
import torch
from prototorch.functions import (activations, competitions, distances,
initializers, losses)
@@ -139,7 +138,7 @@ class TestCompetitions(unittest.TestCase):
def test_knnc_k1(self):
d = torch.tensor([[2.0, 3.0, 1.99, 3.01], [2.0, 3.0, 2.01, 3.0]])
labels = torch.tensor([0, 1, 2, 3])
actual = competitions.knnc(d, labels, k=torch.tensor([1]))
actual = competitions.knnc(d, labels, k=1)
desired = torch.tensor([2, 0])
mismatch = np.testing.assert_array_almost_equal(actual,
desired,