参考文章:
https://zhuanlan.zhihu.com/p/117634492
DTW更加适合在时间轴上有伸缩的情况,在语音序列匹配中使用广泛。
m×n
里面存放的值,比如说是 [i, j]
应该是序列a中第i个值和序列b中第j个值的距离。假设:
(1)有序列a长度为m
(2)有序列b长度为n
(3)m×n 的矩阵 叫做d[ ][ ]
(4)这里定义两个点之间的距离,就是差的绝对值,叫做fun()
计算d[0][0] = fun(a[0], b[0])
计算第0行
d[i][0] = fun(a[i],b[0])
计算第0列
dp[0][j] = fun(a[0],b[j])
计算剩余的值:选择{左边、上边、左上角}三个值中最小的值 + fun (a[i], b[j] )
d[i][j] = min(d[i-1][j-1],d[i-1][j],dd[i][j-1]) + fun(a[i],b[j])
# 计算序列组成单元之间的距离,可以是欧氏距离,也可以是任何其他定义的距离,这里使用绝对值
def distance(w1,w2):
d = abs(w2 - w1)
return d
# DTW计算序列s1,s2的最小距离
def DTW(s1,s2):
m = len(s1)
n = len(s2)
# 构建二位dp矩阵,存储对应每个子问题的最小距离
dp = [[0]*n for _ in range(m)]
# 起始条件,计算单个字符与一个序列的距离
for i in range(m):
dp[i][0] = distance(s1[i],s2[0])
for j in range(n):
dp[0][j] = distance(s1[0],s2[j])
# 利用递推公式,计算每个子问题的最小距离,矩阵最右下角的元素即位最终两个序列的最小值
for i in range(1,m):
for j in range(1,n):
dp[i][j] = min(dp[i-1][j-1],dp[i-1][j],dp[i][j-1]) + distance(s1[i],s2[j])
return dp[-1][-1]
s1 = [1,3,2,4,2]
s2 = [0,3,4,2,2]
print('DTW distance: ',DTW(s1,s2)) # 输出 DTW distance: 2