prepare merge
This commit is contained in:
parent
9dac6e9612
commit
de03a9da58
@ -848,38 +848,42 @@ set_tunable(I,Slope,P) :-
|
|||||||
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
||||||
user:evaluate(LLH_Training_Queries, X,Grad,N,_,_) :-
|
user:evaluate(LLH_Training_Queries, X,Grad,N,_,_) :-
|
||||||
%Handle = user_error,
|
%Handle = user_error,
|
||||||
GradCount <== array[N] of ints,
|
example_count(TrainingExampleCount),
|
||||||
|
LLs <== array[TrainingExampleCount ] of floats,
|
||||||
Probs <== array[N] of floats,
|
Probs <== array[N] of floats,
|
||||||
problog_flag(sigmoid_slope,Slope),
|
problog_flag(sigmoid_slope,Slope),
|
||||||
N1 is N-1,
|
N1 is N-1,
|
||||||
|
|
||||||
forall(between(0,N1,I),
|
forall(between(0,N1,I),
|
||||||
(Grad[I] <== 0.0, S <== X[I], sigmoid(S,Slope, P), Probs[I] <== P)
|
(Grad[I] <== 0.0, S <== X[I], sigmoid(S,Slope, P), Probs[I] <== P)
|
||||||
|
), nl,
|
||||||
|
forall(
|
||||||
|
full_example(QueryID,QueryProb,BDD),
|
||||||
|
compute_grad(QueryID, BDD, QueryProb,Grad, Probs, Slope,LLs)
|
||||||
),
|
),
|
||||||
findall(LL,
|
LLH_Training_Queries <== sum(LLs),
|
||||||
compute_grad(Grad, GradCount, Probs, Slope, LL),
|
writeln(LLH_Training_Queries).
|
||||||
LLs
|
|
||||||
),
|
|
||||||
sum_list(LLs,LLH_Training_Queries).
|
|
||||||
%wrap(X, Grad, GradCount).
|
%wrap(X, Grad, GradCount).
|
||||||
|
|
||||||
|
full_example(QueryID,QueryProb,BDD) :-
|
||||||
compute_grad(Grad, GradCount, Probs, Slope, LL) :-
|
|
||||||
user:example(QueryID,_Query,QueryProb,_),
|
user:example(QueryID,_Query,QueryProb,_),
|
||||||
recorded(QueryID,BDD,_),
|
recorded(QueryID,BDD,_),
|
||||||
BDD = bdd(_Dir, _GradTree, MapList),
|
BDD = bdd(_Dir, _GradTree, MapList),
|
||||||
MapList = [_|_],
|
MapList = [_|_].
|
||||||
|
|
||||||
|
compute_grad(QueryID,BDD,QueryProb, Grad, Probs, Slope, LLs) :-
|
||||||
|
BDD = bdd(_Dir, _GradTree, MapList),
|
||||||
bind_maplist(MapList, Slope, Probs),
|
bind_maplist(MapList, Slope, Probs),
|
||||||
%writeln( MapList ),
|
recorded(QueryID,BDD,_),
|
||||||
qprobability(BDD,Slope,BDDProb),
|
qprobability(BDD,Slope,BDDProb),
|
||||||
LL is (((BDDProb)-(QueryProb))**2),
|
LL is (BDDProb-QueryProb)*(BDDProb-QueryProb),
|
||||||
|
LLs[QueryID] <== LL,
|
||||||
%writeln( qprobability(BDD,Slope,BDDProb) ),
|
%writeln( qprobability(BDD,Slope,BDDProb) ),
|
||||||
forall(
|
forall(
|
||||||
member(I-_, MapList),
|
member(I-_, MapList),
|
||||||
gradientpair(I, BDD,Slope,BDDProb, QueryProb, Grad, Probs, GradCount)
|
gradientpair(I, BDD,Slope,BDDProb, QueryProb, Grad, Probs)
|
||||||
).
|
).
|
||||||
|
|
||||||
gradientpair(I, BDD,Slope,BDDProb, QueryProb, Grad, Probs, GradCount) :-
|
gradientpair(I, BDD,Slope,BDDProb, QueryProb, Grad, Probs) :-
|
||||||
qgradient(I, BDD, Slope, FactID, GradValue),
|
qgradient(I, BDD, Slope, FactID, GradValue),
|
||||||
% writeln(FactID),
|
% writeln(FactID),
|
||||||
G0 <== Grad[FactID],
|
G0 <== Grad[FactID],
|
||||||
@ -887,9 +891,6 @@ gradientpair(I, BDD,Slope,BDDProb, QueryProb, Grad, Probs, GradCount) :-
|
|||||||
%writeln( GN is G0-GradValue*(QueryProb-BDDProb)),
|
%writeln( GN is G0-GradValue*(QueryProb-BDDProb)),
|
||||||
GN is G0-GradValue*2*Prob*(1-Prob)*(QueryProb-BDDProb),
|
GN is G0-GradValue*2*Prob*(1-Prob)*(QueryProb-BDDProb),
|
||||||
%writeln(FactID:(G0->GN)),
|
%writeln(FactID:(G0->GN)),
|
||||||
GC <== GradCount[FactID],
|
|
||||||
GC1 is GC+1,
|
|
||||||
GradCount[FactID] <== GC1,
|
|
||||||
Grad[FactID] <== GN.
|
Grad[FactID] <== GN.
|
||||||
|
|
||||||
qprobability(bdd(Dir, Tree, _MapList), Slope, Prob) :-
|
qprobability(bdd(Dir, Tree, _MapList), Slope, Prob) :-
|
||||||
|
@ -835,5 +835,5 @@ class YAPRun:
|
|||||||
(query, _,loop, sols) = self.clean_end(query)
|
(query, _,loop, sols) = self.clean_end(query)
|
||||||
return (program, query, loop, sols)
|
return (program, query, loop, sols)
|
||||||
|
|
||||||
global
|
# global
|
||||||
globals = {}
|
#globals = {}
|
||||||
|
Reference in New Issue
Block a user