about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/SLP/misc/dag_to_cpdag1.m
blob: c3189a1d93d1a4c27e96de7cc3e166187ad89aca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
function  [cpdag] = dag_to_cpdag1(dags)
% 2
% (also works with a cell array of dags, returning a cell array of cpdags)
% DAG_TO_CPDAG produce a N*N matrix which values respect :
%
% 	If the edge is compelled then 1 on the edge.
%	If the edge is reversible then 1 on the edge and 1 in the reverse edge.
%
% Make sure that the entry is a DAG.
%
% See D.M. Chickering: "Learning Equivalence Classes of Bayesian-Network Structures".
%
% 
% francois.olivier.c.h@gmail.com, philippe.leray@univ-nantes.fr

if ~iscell(dags)
    dag=cell(1,1);
    dag{1}=dags;
else
    dag=dags;
end

for da=1:length(dag)
    cpdags{da} = abs(label_edges(dag{da}));
end

if ~iscell(dags)
    cpdag=cpdags{1};
else
    cpdag=cpdags;
end

%%==============================================================================

function [label] = label_edges(dag)
% LABEL-EDGES produce a N*N matrix which values are
% 	+1 if the edge is compelled or
%	-1 if the edge is reversible.
% Make sure that the entry is a DAG.
%
% francois.olivier.c.h@gmail.com

N=length(dag);
[order xedge yedge] = order_edges(dag);
label = 2*dag; % all edges as unknown

NbEdges = length(xedge) ;
%xedge=x, yedge=y, %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%


for Edge=1:NbEdges,
    xlow=xedge(Edge);
    ylow=yedge(Edge);
    if label(xlow,ylow)==2
        fin = 0;

        %wcompelled = find(label(:,xlow)~=2);
        wcompelled = find(label(:,xlow)==1);

        parenty = find(label(:,ylow)~=0);
        %sonsy = find(label(ylow,:)~=0);

        for s = 1:length(wcompelled)
           w = wcompelled(s);
           if ~ismember(w,parenty)
                label(xlow,ylow)=1; 
                label(ylow,xlow)=0; %
                label(parenty,ylow)=1;
                label(ylow,parenty)=0; %
                %label(ylow,sonsy)=1; %
                %label(sonsy,ylow)=0; %
                fin = 1;

            elseif fin == 0
                label(w,ylow)=1;
                %label(ylow,w)=0; %
            end
        end
        if fin == 0
            parentx = [xlow ; find(label(:,xlow)~=0)];
            if ~isempty(mysetdiff(parenty,parentx))
                %label(xlow,ylow)=1; %
                %label(ylow,xlow)=0; %

                label(find(label(:,ylow)==2),ylow)=1;
                label(ylow,find(label(ylow,:)==2))=1; %
            else	
                label(xlow,ylow)=-1;
                label(ylow,xlow)=-1; %
                ttp = find(label(:,ylow)==2);
                label(ttp,ylow)=-1;
                label(ylow,ttp)=-1; %
            end
        end
    end
end

%%%========================================================================================
function [order, x, y] = order_edges(dag)
% ORDER_EDGES produce a total (natural) ordering over the edges in a DAG.
% Make sure that the entry is a DAG.
%
% francois.olivier.c.h@gmail.com
%
% 2 mai 2003

if acyclic(dag)==0
    error('Requires an acyclic graph');
end

N=length(dag);
order = zeros(N,N);

node_order = topological_sort(dag);
[tmp oo] = sort(node_order);

dag2=dag(node_order,node_order);
[x y]=find(flipud(dag2)==1);
nb_edges=length(x);

if nb_edges~=0
  order(sub2ind([N N],N+1-x,y))=1:nb_edges ;
end

order=order(oo,oo);
x=node_order(N+1-x);
y=node_order(y);