from __future__ import division, print_function import unittest import numpy as np from smqtk.representation.descriptor_element.local_elements import \ DescriptorMemoryElement from smqtk.algorithms.relevancy_index.libsvm_hik import LibSvmHikRelevancyIndex if LibSvmHikRelevancyIndex.is_usable(): class TestIqrSvmHik(unittest.TestCase): @classmethod def setUpClass(cls): # Don't need to clear cache because we're setting the vectors here cls.d0 = DescriptorMemoryElement('index', 0) cls.d0.set_vector(np.array([1, 0, 0, 0, 0], float)) cls.d1 = DescriptorMemoryElement('index', 1) cls.d1.set_vector(np.array([0, 1, 0, 0, 0], float)) cls.d2 = DescriptorMemoryElement('index', 2) cls.d2.set_vector(np.array([0, 0, 1, 0, 0], float)) cls.d3 = DescriptorMemoryElement('index', 3) cls.d3.set_vector(np.array([0, 0, 0, 1, 0], float)) cls.d4 = DescriptorMemoryElement('index', 4) cls.d4.set_vector(np.array([0, 0, 0, 0, 1], float)) cls.d5 = DescriptorMemoryElement('index', 5) cls.d5.set_vector(np.array([0.5, 0, 0.5, 0, 0], float)) cls.d6 = DescriptorMemoryElement('index', 6) cls.d6.set_vector(np.array([.2, .2, .2, .2, .2], float)) cls.index_descriptors = [ cls.d0, cls.d1, cls.d2, cls.d3, cls.d4, cls.d5, cls.d6
from __future__ import division, print_function import unittest import numpy as np from smqtk.representation.descriptor_element.local_elements import \ DescriptorMemoryElement from smqtk.algorithms.relevancy_index.libsvm_hik import LibSvmHikRelevancyIndex if LibSvmHikRelevancyIndex.is_usable(): class TestIqrSvmHik (unittest.TestCase): @classmethod def setUpClass(cls): # Don't need to clear cache because we're setting the vectors here cls.d0 = DescriptorMemoryElement('index', 0) cls.d0.set_vector(np.array([1, 0, 0, 0, 0], float)) cls.d1 = DescriptorMemoryElement('index', 1) cls.d1.set_vector(np.array([0, 1, 0, 0, 0], float)) cls.d2 = DescriptorMemoryElement('index', 2) cls.d2.set_vector(np.array([0, 0, 1, 0, 0], float)) cls.d3 = DescriptorMemoryElement('index', 3) cls.d3.set_vector(np.array([0, 0, 0, 1, 0], float)) cls.d4 = DescriptorMemoryElement('index', 4) cls.d4.set_vector(np.array([0, 0, 0, 0, 1], float)) cls.d5 = DescriptorMemoryElement('index', 5) cls.d5.set_vector(np.array([0.5, 0, 0.5, 0, 0], float)) cls.d6 = DescriptorMemoryElement('index', 6) cls.d6.set_vector(np.array([.2, .2, .2, .2, .2], float))