我想在Matlab中高效地计算通用核函数的核矩阵。这意味着我需要计算k(x,y)
,其中x
是X
的每一行,y
是Y
的每一行。以下是实现我所需功能的Matlab代码,但它的速度相当慢,
function K=compute_kernel( k_func, X, Y ) m = size(X,1); n = size(Y,1); K = zeros(m,n); for i = 1:m for j = 1:n K(i,j) = k_func(X(i,:)', Y(j,:)'); end endend
有没有其他方法来解决这个问题,例如使用bsxfun
的变体,在X
和Y
的每一行上调用函数?
回答:
遗憾的是,pdist2(X,Y, dist_func)
计算的是dist_func(X, Y(i,:))
,而不是dist_func(X(i,:), Y(i,:))
。所以我真正需要的函数是,
function K=compute_kernel( k_func, X, Y ) % 哇!高效计算核函数的方法 size(X) size(Y) m = size(X,1); n = size(Y,1); for i = 1:n K(:,i) = pdist2( Y(i,:), X, k_func);end
虽然没有直接使用pdist2
那么简洁,但仍然比之前的方法高效得多。