DCausse has uploaded a new change for review. (
https://gerrit.wikimedia.org/r/367899 )
Change subject: Minor fixes in training_pipeline
......................................................................
Minor fixes in training_pipeline
Change-Id: I808539c0eab690c1edb534ec9ed6b52418657b78
---
M mjolnir/cli/training_pipeline.py
1 file changed, 2 insertions(+), 2 deletions(-)
git pull ssh://gerrit.wikimedia.org:29418/search/MjoLniR
refs/changes/99/367899/1
diff --git a/mjolnir/cli/training_pipeline.py b/mjolnir/cli/training_pipeline.py
index fc08b8d..ca984ee 100644
--- a/mjolnir/cli/training_pipeline.py
+++ b/mjolnir/cli/training_pipeline.py
@@ -55,8 +55,8 @@
num_cv_jobs=num_cv_jobs, num_workers=num_workers,
target_node_evaluations=target_node_evaluations)
- print 'CV test-ndcg@10: %.4f' % (tune_results['metrics']['test'])
- print 'CV train-ndcg@10: %.4f' % (tune_results['metrics']['train'])
+ print 'CV test-ndcg@10: %.4f' % (tune_results['metrics']['cv-test'])
+ print 'CV train-ndcg@10: %.4f' % (tune_results['metrics']['cv-train'])
# Train a model over all data with best params
best_params = tune_results['params']
--
To view, visit https://gerrit.wikimedia.org/r/367899
To unsubscribe, visit https://gerrit.wikimedia.org/r/settings
Gerrit-MessageType: newchange
Gerrit-Change-Id: I808539c0eab690c1edb534ec9ed6b52418657b78
Gerrit-PatchSet: 1
Gerrit-Project: search/MjoLniR
Gerrit-Branch: master
Gerrit-Owner: DCausse <[email protected]>
_______________________________________________
MediaWiki-commits mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/mediawiki-commits
