GitHub/gf-core
1
0
Fork 1
mirror of https://github.com/GrammaticalFramework/gf-core.git synced 2026-09-10 20:36:01 -06:00
Code Issues Packages Projects Releases Wiki Activity

make sure that we don't miss low-prob trees

Browse Source
This commit is contained in:
Krasimir Angelov
2026-08-13 08:11:45 +02:00
parent 5b30a80e3f
commit 1a0a7f9d09
1 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/runtime/c/pgf/parser.cxx
+19 -1
View File
@@ -1471,7 +1471,7 @@ void PgfParser::final_item(State *state, CCat *ccat, Item *item, interval_t valu
estate->expr = 0; estate->expr = 0;
estate->prob = 0; estate->prob = 0;
estate->hash = 0; estate->hash = 0;
estate->res = NULL; estate->res = ccat;
estate->index = 0; estate->index = 0;
estate->n_args = item->args.size(); estate->n_args = item->args.size();
for (size_t i = 0; i < estate->n_args; i++) { for (size_t i = 0; i < estate->n_args; i++) {
@@ -1480,6 +1480,24 @@ void PgfParser::final_item(State *state, CCat *ccat, Item *item, interval_t valu
} }
queue.push_back(estate); queue.push_back(estate);
std::push_heap(queue.begin(), queue.end(), estate_comp); std::push_heap(queue.begin(), queue.end(), estate_comp);
} else if (ccat != NULL && ccat->pending.size() > 0) {
auto lin = ref<PgfConcrLin>::untagged(item->rule->container);
ExprState *estate = new(item->args.size()) ExprState;
estate->expr = u->efun(&lin->name);
estate->prob = ccat->pending[0]->prob-ccat->viterbi_prob+lin->absfun->prob;
estate->hash = 0;
estate->res = ccat;
estate->index = 0;
estate->n_args = item->args.size();
for (size_t i = 0; i < lin->name.size; i++) {
estate->hash = estate->hash * 101 + lin->name.text[i];
}
for (size_t i = 0; i < estate->n_args; i++) {
estate->args[i] = item->args[i];
estate->prob += estate->args[i]->viterbi_prob;
}
queue.push_back(estate);
std::push_heap(queue.begin(), queue.end(), estate_comp);
} }
} }