2012-06-17 6 views
8

pierwszego rzędu macierzy przejścia z 6 państw może być constructed very elegantly as następującoKonstruowanie wielo-order Markowa macierzy przejścia łańcucha w Matlab

x = [1 6 1 6 4 4 4 3 1 2 2 3 4 5 4 5 2 6 2 6 2 6]; % the Markov chain 
tm = full(sparse(x(1:end-1),x(2:end),1)) % the transition matrix. 

Więc tutaj jest mój problem, w jaki sposób skonstruować drugiego rzędu macierzy przejścia elegancko? Rozwiązanie wymyśliłem jest następujący

[si sj] = ndgrid(1:6); 
s2 = [si(:) sj(:)]; % combinations for 2 contiguous states 
tm2 = zeros([numel(si),6]); % initialize transition matrix 
for i = 3:numel(x) % construct transition matrix 
    tm2(strmatch(num2str(x(i-2:i-1)),num2str(s2)),x(i))=... 
    tm2(strmatch(num2str(x(i-2:i-1)),num2str(s2)),x(i))+1; 
end 

Czy istnieje jeden/dwa-liner, bez pętli alternatywa?

-

Edycja: że próbował porównanie mi roztworu względem Amro "X = zaokrąglenie (5 * rand ([1,1000]) + 1);"

% ted teng's solution 
Elapsed time is 2.225573 seconds. 
% Amro's solution 
Elapsed time is 0.042369 seconds. 

Co za różnica! FYI, grp2idx jest dostępna online.

Odpowiedz

8

Spróbuj wykonać następujące czynności:

%# sequence of states 
x = [1 6 1 6 4 4 4 3 1 2 2 3 4 5 4 5 2 6 2 6 2 6]; 
N = max(x); 

%# extract contiguous sequences of 2 items from the above 
bigrams = cellstr(num2str([x(1:end-2);x(2:end-1)]')); 

%# all possible combinations of two symbols 
[X,Y] = ndgrid(1:N,1:N); 
xy = cellstr(num2str([X(:),Y(:)])); 

%# map bigrams to numbers starting from 1 
[g,gn] = grp2idx([xy;bigrams]); 
s1 = g(N*N+1:end); 

%# items following the bigrams 
s2 = x(3:end); 

%# transition matrix 
tm = full(sparse(s1,s2,1,N*N,N)); 
spy(tm) 

transition matrix

+0

Panie, jesteś królem Matlab @ Stos Exchange. Nawet w niedziele! – teng

+2

@tedteng: lol thanks :) Funkcja [GRP2IDX] (http://www.mathworks.com/help/toolbox/stats/grp2idx.html) jest częścią zestawu narzędzi statystycznych, ale można ją zastąpić przez [UNIQUE ] (http://www.mathworks.com/help/techdoc/ref/unique.html): '[gn, ~, g] = unique ([xy; bigrams], 'stable');' – Amro

+0

Czy to będzie łatwo rozszerzyć tę metodę na dwuwymiarowy ** x **? – HCAI

Powiązane problemy