Java实现DTW

1、DTW类

Dtw.java

-----------------------------------------------------------------

public class Dtw {

public double getMin(double a, double b, double c) {
double min = a;
if (b > a)
min = a;
else if (c > b) {
min = b;
} else {
min = c;
}
return min;
}

public double getDistance(double[] seqa, double[] seqb) {
double distance = 0;
int lena = seqa.length;
int lenb = seqb.length;
double[][] c = new double[lena][lenb];
for (int i = 0; i < lena; i++) {
for (int j = 0; j < lenb; j++) {
c[i][j] = 1;
}
}
for (int i = 0; i < lena; i++) {
for (int j = 0; j < lenb; j++) {
double tmp = (seqa[i] - seqb[j]) * (seqa[i] - seqb[j]);
if (j == 0 && i == 0)
c[i][j] = tmp;
else if (j > 0)
c[i][j] = c[i][j - 1] + tmp;
if (i > 0) {
if (j == 0)
c[i][j] = tmp + c[i - 1][j];
else
c[i][j] = tmp + getMin(c[i][j - 1], c[i - 1][j - 1], c[i - 1][j]);
}
}
}
distance = c[lena - 1][lenb - 1];
return distance;
}

}

2、测试

Test.java

-----------------------------------------------------------------

public class Test {
public static void main(String[] args) {
Dtw dtw = new Dtw();
double[] x = { 3, 5, 6, 7, 7, 1 };
double[] y = { 3, 6, 6, 7, 8, 1, 1 };
double[] z = { 2, 5, 7, 7, 7, 7, 2 };
System.out.println(dtw.getDistance(x, y));
System.out.println(dtw.getDistance(x, z));
}
}