Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011 #include <shogun/lib/common.h>
00012 #include <shogun/mathematics/Math.h>
00013 #include <shogun/kernel/TensorProductPairKernel.h>
00014 #include <shogun/io/SGIO.h>
00015
00016 using namespace shogun;
00017
00018 CTensorProductPairKernel::CTensorProductPairKernel()
00019 : CDotKernel(0), subkernel(NULL)
00020 {
00021 register_params();
00022 }
00023
00024 CTensorProductPairKernel::CTensorProductPairKernel(int32_t size, CKernel* s)
00025 : CDotKernel(size), subkernel(s)
00026 {
00027 SG_REF(subkernel);
00028 register_params();
00029 }
00030
00031 CTensorProductPairKernel::CTensorProductPairKernel(CDenseFeatures<int32_t>* l, CDenseFeatures<int32_t>* r, CKernel* s)
00032 : CDotKernel(10), subkernel(s)
00033 {
00034 SG_REF(subkernel);
00035 init(l, r);
00036 register_params();
00037 }
00038
00039 CTensorProductPairKernel::~CTensorProductPairKernel()
00040 {
00041 SG_UNREF(subkernel);
00042 cleanup();
00043 }
00044
00045 bool CTensorProductPairKernel::init(CFeatures* l, CFeatures* r)
00046 {
00047 CDotKernel::init(l, r);
00048 init_normalizer();
00049 return true;
00050 }
00051
00052 float64_t CTensorProductPairKernel::compute(int32_t idx_a, int32_t idx_b)
00053 {
00054 int32_t alen, blen;
00055 bool afree, bfree;
00056
00057 int32_t* avec=((CDenseFeatures<int32_t>*) lhs)->get_feature_vector(idx_a, alen, afree);
00058 int32_t* bvec=((CDenseFeatures<int32_t>*) rhs)->get_feature_vector(idx_b, blen, bfree);
00059
00060 ASSERT(alen==2);
00061 ASSERT(blen==2);
00062
00063 CKernel* k=subkernel;
00064 ASSERT(k && k->has_features());
00065
00066 int32_t a=avec[0];
00067 int32_t b=avec[1];
00068 int32_t c=bvec[0];
00069 int32_t d=bvec[1];
00070
00071 float64_t result = k->kernel(a,c)*k->kernel(b,d) + k->kernel(a,d)*k->kernel(b,c);
00072
00073 ((CDenseFeatures<int32_t>*) lhs)->free_feature_vector(avec, idx_a, afree);
00074 ((CDenseFeatures<int32_t>*) rhs)->free_feature_vector(bvec, idx_b, bfree);
00075
00076 return result;
00077 }
00078
00079 void CTensorProductPairKernel::register_params()
00080 {
00081 SG_ADD((CSGObject**)&subkernel, "subkernel", "the subkernel", MS_AVAILABLE);
00082 }