diff options
author | Chris Dyer <cdyer@cs.cmu.edu> | 2012-06-24 22:30:50 -0400 |
---|---|---|
committer | Chris Dyer <cdyer@cs.cmu.edu> | 2012-06-24 22:30:50 -0400 |
commit | ba23531ee5ba0c281ddfe1b40e5148ae2d94c277 (patch) | |
tree | 4bba75548be6bfb62480ad842d5618fdba5aaf00 /training/risk.cc | |
parent | d19ddda6446d3093de1617027350d34df20a1b50 (diff) |
minimum risk training, not completely ready for primetime
Diffstat (limited to 'training/risk.cc')
-rw-r--r-- | training/risk.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/training/risk.cc b/training/risk.cc index 347ed3cb..d5a12cfd 100644 --- a/training/risk.cc +++ b/training/risk.cc @@ -31,7 +31,9 @@ double CandidateSetRisk::operator()(const vector<double>& params, for (unsigned i = 0; i < cands_.size(); ++i) { const double log_prob = cands_[i].fmap.dot(params) - log_z; const double prob = exp(log_prob); - const double r = prob * metric_.ComputeScore(cands_[i].eval_feats); + const double cost = metric_.IsErrorMetric() ? metric_.ComputeScore(cands_[i].eval_feats) + : 1.0 - metric_.ComputeScore(cands_[i].eval_feats); + const double r = prob * cost; risk += r; if (g) (*g) += (cands_[i].fmap - exp_feats) * r; } |