annotate maths/KLDivergence.cpp @ 256:43943a4382ef

* Add cosine distance and the self-similarity matrix used for SB rhythmic similarity * Pull out SB timbral similarity KL divergence into its own file
author Chris Cannam <c.cannam@qmul.ac.uk>
date Fri, 18 Jan 2008 14:40:20 +0000
parents
children 499d438b52ba
rev   line source
c@256 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
c@256 2
c@256 3 /*
c@256 4 QM DSP Library
c@256 5
c@256 6 Centre for Digital Music, Queen Mary, University of London.
c@256 7 This file copyright 2008 QMUL
c@256 8 All rights reserved.
c@256 9 */
c@256 10
c@256 11 #include "KLDivergence.h"
c@256 12
c@256 13 double KLDivergence::distance(const vector<double> &m1,
c@256 14 const vector<double> &v1,
c@256 15 const vector<double> &m2,
c@256 16 const vector<double> &v2)
c@256 17 {
c@256 18 int sz = m1.size();
c@256 19
c@256 20 double d = -2.0 * sz;
c@256 21
c@256 22 for (int k = 0; k < sz; ++k) {
c@256 23 d += v1[k] / v2[k] + v2[k] / v1[k];
c@256 24 d += (m1[k] - m2[k]) * (1.0 / v1[k] + 1.0 / v2[k]) * (m1[k] - m2[k]);
c@256 25 }
c@256 26
c@256 27 d /= 2.0;
c@256 28
c@256 29 return d;
c@256 30 }