afb734b51a
git-svn-id: https://yap.svn.sf.net/svnroot/yap/trunk@2093 b08c6af1-5177-4d33-ba66-4b1c6b8b522a
43 lines
1006 B
Prolog
43 lines
1006 B
Prolog
%
|
|
% This assumes we have a single big example.
|
|
%
|
|
|
|
:- use_module(library('clpbn_learning/utils'),
|
|
[run_all/1,
|
|
clpbn_vars/2]).
|
|
|
|
:- module(bnt_mle, [learn_parameters/2]).
|
|
|
|
%
|
|
% full evidence learning
|
|
%
|
|
learn_parameters(Items, Tables, Extras) :-
|
|
run_all(Items),
|
|
attributes:all_attvars(AVars),
|
|
% sort and incorporte evidence
|
|
clpbn_vars(AVars, AllVars),
|
|
mk_sample(AllVars, Sample),
|
|
compute_tables(Extras, Sample, Tables).
|
|
|
|
mk_sample(AllVars, NVars, LL) :-
|
|
add2sample(AllVars, Sample),
|
|
msort(Sample, AddL),
|
|
compute_params(AddL, Parms).
|
|
|
|
add2sample([], []).
|
|
add2sample([V|Vs],[val(Id,[Ev|EParents])|Vals]) :-
|
|
clpbn:get_atts(V, [evidence(Ev),dist(Id,Parents)]),
|
|
get_eparents(Parents, EParents),
|
|
add2sample(Vs, Vals).
|
|
|
|
get_eparents([P|Parents], [E|EParents]) :-
|
|
clpbn:get_atts(V, [evidence(Ev)]),
|
|
get_eparents(Parents, EParents).
|
|
get_eparents([], []).
|
|
|
|
|
|
compute_tables([], Sample, Tables) :-
|
|
mle(Sample, Tables).
|
|
compute_tables([laplace|_], Sample, Tables) :-
|
|
laplace(Sample, Tables).
|