Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011 #include "lib/config.h"
00012 #include "lib/common.h"
00013 #include "lib/io.h"
00014 #include "distance/JensenMetric.h"
00015 #include "features/Features.h"
00016 #include "features/SimpleFeatures.h"
00017
00018 using namespace shogun;
00019
00020 CJensenMetric::CJensenMetric()
00021 : CSimpleDistance<float64_t>()
00022 {
00023 }
00024
00025 CJensenMetric::CJensenMetric(CSimpleFeatures<float64_t>* l, CSimpleFeatures<float64_t>* r)
00026 : CSimpleDistance<float64_t>()
00027 {
00028 init(l, r);
00029 }
00030
00031 CJensenMetric::~CJensenMetric()
00032 {
00033 cleanup();
00034 }
00035
00036 bool CJensenMetric::init(CFeatures* l, CFeatures* r)
00037 {
00038 bool result=CSimpleDistance<float64_t>::init(l,r);
00039
00040 return result;
00041 }
00042
00043 void CJensenMetric::cleanup()
00044 {
00045 }
00046
00047 float64_t CJensenMetric::compute(int32_t idx_a, int32_t idx_b)
00048 {
00049 int32_t alen, blen;
00050 bool afree, bfree;
00051
00052 float64_t* avec=
00053 ((CSimpleFeatures<float64_t>*) lhs)->get_feature_vector(idx_a, alen, afree);
00054 float64_t* bvec=
00055 ((CSimpleFeatures<float64_t>*) rhs)->get_feature_vector(idx_b, blen, bfree);
00056
00057 ASSERT(alen==blen);
00058
00059 float64_t absTmp = 0;
00060 float64_t result=0;
00061 {
00062 for (int32_t i=0; i<alen; i++)
00063 {
00064 absTmp=0.5*(avec[i]+bvec[i]);
00065 if(avec[i]>0)
00066 result+=avec[i]*log(avec[i]/absTmp);
00067 if(bvec[i]>0)
00068 result+=bvec[i]*log(bvec[i]/absTmp);
00069 }
00070
00071 }
00072
00073 ((CSimpleFeatures<float64_t>*) lhs)->free_feature_vector(avec, idx_a, afree);
00074 ((CSimpleFeatures<float64_t>*) rhs)->free_feature_vector(bvec, idx_b, bfree);
00075
00076
00077 return result;
00078 }