Chromium Code Reviews| Index: appengine/findit/crash/scorers/aggregated_scorer.py |
| diff --git a/appengine/findit/crash/scorers/aggregated_scorer.py b/appengine/findit/crash/scorers/aggregated_scorer.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..b1c89e460f84f4dc4ff01856aeaab7c340df6d02 |
| --- /dev/null |
| +++ b/appengine/findit/crash/scorers/aggregated_scorer.py |
| @@ -0,0 +1,34 @@ |
| +# Copyright 2016 The Chromium Authors. All rights reserved. |
| +# Use of this source code is governed by a BSD-style license that can be |
| +# found in the LICENSE file. |
| + |
| +"""Uses aggregators to aggregator results from different scorers.""" |
| + |
| +from crash.scorers import aggregators |
| + |
| + |
| +class AggregatedScorer(object): |
| + |
| + def __init__(self, scorers): |
| + self.scorers = scorers |
| + |
| + def Score(self, result, |
| + score_aggregator=aggregators.Multiplier(), |
| + reasons_aggregator=aggregators.IdentityAggregator(), |
| + changed_files_aggregator=aggregators.ChangedFilesAggregator()): |
| + """Aggregates score, reasons and changed_files from all the scorers. |
| + |
| + Note: This method sets confidence, reasons and changed_files of results. |
| + """ |
| + # Transforms array of [(score1, reason1, changed_files1), (score2, reason2, |
| + # changed_files2)] to [(score1, score2), (reason1, reason2), |
| + # (changed_files1, changed_files2)] for aggregators to aggregate. |
| + scores, reasons, changed_files = zip(*[ |
| + scorer(result) for scorer in self.scorers]) |
| + |
| + # Only keep the first 2 digits after decimal point. |
|
lijeffrey
2016/07/19 18:31:30
nit: It's not obvious the code here does the 2 dig
Sharu Jiang
2016/07/21 03:30:24
Done.
|
| + result.confidence = score_aggregator(list(scores)) |
| + result.reasons = reasons_aggregator(list(reasons)) |
| + result.changed_files = changed_files_aggregator(list(changed_files)) |
| + |
| + return result.confidence, result.reasons, result.changed_files |