Exemplo n.º 1
0
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
Exemplo n.º 2
0
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))