diff options
author | Jake Hofman <jhofman@gmail.com> | 2014-01-01 01:37:38 +0400 |
---|---|---|
committer | Jake Hofman <jhofman@gmail.com> | 2014-01-01 01:37:38 +0400 |
commit | 93819c84faacdf5f0f187dd71542491a19d33568 (patch) | |
tree | 66f1bd4d767e372191ef3bd93d3f2ae1d80a40c9 /vowpalwabbit/gd_mf.cc | |
parent | 4bca4ce57e90fbd9af8f50f0fba994244b5ae2e7 (diff) |
cleaned up base learner predict functions
Diffstat (limited to 'vowpalwabbit/gd_mf.cc')
-rw-r--r-- | vowpalwabbit/gd_mf.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/vowpalwabbit/gd_mf.cc b/vowpalwabbit/gd_mf.cc index 4edcb8ca..5230b8e6 100644 --- a/vowpalwabbit/gd_mf.cc +++ b/vowpalwabbit/gd_mf.cc @@ -293,10 +293,9 @@ void end_pass(void* d) // placeholder void predict(void* d, learner& base, example* ec) { - bool test_only = ec->test_only; - ec->test_only = true; - learn(d, base, ec); - ec->test_only = test_only; + vw* all = ((gdmf*)d)->all; + + mf_predict(*all,ec); } learner* setup(vw& all) |