diff options
author | fenchel <fenchel@squidhive.net> | 2014-07-15 18:58:01 +0400 |
---|---|---|
committer | fenchel <fenchel@squidhive.net> | 2014-07-15 18:58:01 +0400 |
commit | 4ab098c2c4a43a262574b61b233e40e293551f04 (patch) | |
tree | 23f7dafe08dbae9bbdb6f7414a514e69e276ea15 | |
parent | 9dbafb6086c6b3ccfd2bbc9b1746ed3d3f125f65 (diff) |
stage_poly: update ec predictions after reduction
-rw-r--r-- | vowpalwabbit/stagewise_poly.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/vowpalwabbit/stagewise_poly.cc b/vowpalwabbit/stagewise_poly.cc index 28885e28..80735b71 100644 --- a/vowpalwabbit/stagewise_poly.cc +++ b/vowpalwabbit/stagewise_poly.cc @@ -504,9 +504,8 @@ namespace StagewisePoly poly.original_ec = &ec; synthetic_create(poly, ec, false); base.predict(poly.synth_ec); - label_data *ld = (label_data *) ec.ld; - if (ld->label != FLT_MAX) - ec.loss = poly.all->loss->getLoss(poly.all->sd, ld->prediction, ld->label) * ld->weight; + ec.partial_prediction = poly.synth_ec.partial_prediction; + ec.updated_prediction = poly.synth_ec.updated_prediction; } void learn(stagewise_poly &poly, learner &base, example &ec) @@ -522,7 +521,8 @@ namespace StagewisePoly synthetic_create(poly, ec, training); base.learn(poly.synth_ec); - ec.loss = poly.synth_ec.loss; + ec.partial_prediction = poly.synth_ec.partial_prediction; + ec.updated_prediction = poly.synth_ec.updated_prediction; if (ec.example_counter && poly.batch_sz |