Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(157)

Side by Side Diff: components/translate/core/browser/translate_ranker_impl.cc

Issue 2925733002: Move ranker_model_loader to a new component. (Closed)
Patch Set: Adressing sdefresne's comments. Created 3 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2016 The Chromium Authors. All rights reserved. 1 // Copyright 2016 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "components/translate/core/browser/translate_ranker_impl.h" 5 #include "components/translate/core/browser/translate_ranker_impl.h"
6 6
7 #include <cmath> 7 #include <cmath>
8 8
9 #include "base/bind.h" 9 #include "base/bind.h"
10 #include "base/bind_helpers.h" 10 #include "base/bind_helpers.h"
11 #include "base/command_line.h" 11 #include "base/command_line.h"
12 #include "base/files/file_path.h" 12 #include "base/files/file_path.h"
13 #include "base/files/file_util.h" 13 #include "base/files/file_util.h"
14 #include "base/memory/ptr_util.h" 14 #include "base/memory/ptr_util.h"
15 #include "base/metrics/histogram_macros.h" 15 #include "base/metrics/histogram_macros.h"
16 #include "base/metrics/metrics_hashes.h" 16 #include "base/metrics/metrics_hashes.h"
17 #include "base/profiler/scoped_tracker.h" 17 #include "base/profiler/scoped_tracker.h"
18 #include "base/strings/string_util.h" 18 #include "base/strings/string_util.h"
19 #include "base/task_runner.h" 19 #include "base/task_runner.h"
20 #include "base/threading/thread_task_runner_handle.h" 20 #include "base/threading/thread_task_runner_handle.h"
21 #include "components/machine_intelligence/proto/ranker_model.pb.h"
22 #include "components/machine_intelligence/proto/translate_ranker_model.pb.h"
23 #include "components/machine_intelligence/ranker_model.h"
21 #include "components/metrics/proto/translate_event.pb.h" 24 #include "components/metrics/proto/translate_event.pb.h"
22 #include "components/translate/core/browser/proto/ranker_model.pb.h" 25 #include "components/translate/core/browser/translate_download_manager.h"
23 #include "components/translate/core/browser/proto/translate_ranker_model.pb.h"
24 #include "components/translate/core/browser/ranker_model.h"
25 #include "components/translate/core/common/translate_switches.h" 26 #include "components/translate/core/common/translate_switches.h"
26 #include "components/ukm/public/ukm_entry_builder.h" 27 #include "components/ukm/public/ukm_entry_builder.h"
27 #include "components/ukm/public/ukm_recorder.h" 28 #include "components/ukm/public/ukm_recorder.h"
28 #include "components/variations/variations_associated_data.h" 29 #include "components/variations/variations_associated_data.h"
29 #include "url/gurl.h" 30 #include "url/gurl.h"
30 31
31 namespace translate { 32 namespace translate {
32 33
33 namespace { 34 namespace {
34 35
35 using chrome_intelligence::RankerModel; 36 using machine_intelligence::RankerModel;
36 using chrome_intelligence::RankerModelProto; 37 using machine_intelligence::RankerModelProto;
37 using chrome_intelligence::TranslateRankerModel; 38 using machine_intelligence::TranslateRankerModel;
39 using machine_intelligence::RankerModelStatus;
38 40
39 const double kTranslationOfferDefaultThreshold = 0.5; 41 const double kTranslationOfferDefaultThreshold = 0.5;
40 42
41 const char kTranslateRankerModelFileName[] = "Translate Ranker Model"; 43 const char kTranslateRankerModelFileName[] = "Translate Ranker Model";
42 const char kUmaPrefix[] = "Translate.Ranker"; 44 const char kUmaPrefix[] = "Translate.Ranker";
43 const char kUnknown[] = "UNKNOWN"; 45 const char kUnknown[] = "UNKNOWN";
44 46
45 double Sigmoid(double x) { 47 double Sigmoid(double x) {
46 return 1.0 / (1.0 + exp(-x)); 48 return 1.0 / (1.0 + exp(-x));
47 } 49 }
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after
134 ukm::UkmRecorder* ukm_recorder) 136 ukm::UkmRecorder* ukm_recorder)
135 : ukm_recorder_(ukm_recorder), 137 : ukm_recorder_(ukm_recorder),
136 is_logging_enabled_(false), 138 is_logging_enabled_(false),
137 is_query_enabled_(base::FeatureList::IsEnabled(kTranslateRankerQuery)), 139 is_query_enabled_(base::FeatureList::IsEnabled(kTranslateRankerQuery)),
138 is_enforcement_enabled_( 140 is_enforcement_enabled_(
139 base::FeatureList::IsEnabled(kTranslateRankerEnforcement)), 141 base::FeatureList::IsEnabled(kTranslateRankerEnforcement)),
140 is_decision_override_enabled_(base::FeatureList::IsEnabled( 142 is_decision_override_enabled_(base::FeatureList::IsEnabled(
141 translate::kTranslateRankerDecisionOverride)), 143 translate::kTranslateRankerDecisionOverride)),
142 weak_ptr_factory_(this) { 144 weak_ptr_factory_(this) {
143 if (is_query_enabled_ || is_enforcement_enabled_) { 145 if (is_query_enabled_ || is_enforcement_enabled_) {
144 model_loader_ = base::MakeUnique<RankerModelLoader>( 146 model_loader_ = base::MakeUnique<machine_intelligence::RankerModelLoader>(
145 base::Bind(&ValidateModel), 147 base::Bind(&ValidateModel),
146 base::Bind(&TranslateRankerImpl::OnModelAvailable, 148 base::Bind(&TranslateRankerImpl::OnModelAvailable,
147 weak_ptr_factory_.GetWeakPtr()), 149 weak_ptr_factory_.GetWeakPtr()),
148 model_path, model_url, kUmaPrefix); 150 TranslateDownloadManager::GetInstance()->request_context(), model_path,
151 model_url, kUmaPrefix);
149 // Kick off the initial load from cache. 152 // Kick off the initial load from cache.
150 model_loader_->NotifyOfRankerActivity(); 153 model_loader_->NotifyOfRankerActivity();
151 } 154 }
152 } 155 }
153 156
154 TranslateRankerImpl::~TranslateRankerImpl() {} 157 TranslateRankerImpl::~TranslateRankerImpl() {}
155 158
156 // static 159 // static
157 base::FilePath TranslateRankerImpl::GetModelPath( 160 base::FilePath TranslateRankerImpl::GetModelPath(
158 const base::FilePath& data_dir) { 161 const base::FilePath& data_dir) {
(...skipping 210 matching lines...) Expand 10 before | Expand all | Expand 10 after
369 } 372 }
370 } 373 }
371 374
372 } // namespace translate 375 } // namespace translate
373 376
374 std::ostream& operator<<(std::ostream& stream, 377 std::ostream& operator<<(std::ostream& stream,
375 const translate::TranslateRankerFeatures& features) { 378 const translate::TranslateRankerFeatures& features) {
376 features.WriteTo(stream); 379 features.WriteTo(stream);
377 return stream; 380 return stream;
378 } 381 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698