Mercurial > hg > qm-dsp
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 } |