| OLD | NEW |
| 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.h" | 5 #include "components/translate/core/browser/translate_ranker.h" |
| 6 | 6 |
| 7 #include <initializer_list> | 7 #include <initializer_list> |
| 8 #include <memory> | 8 #include <memory> |
| 9 | 9 |
| 10 #include "base/feature_list.h" | 10 #include "base/feature_list.h" |
| (...skipping 148 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 159 double expected = Sigmoid(0.5 * 0.02f + // accept ratio * weight | 159 double expected = Sigmoid(0.5 * 0.02f + // accept ratio * weight |
| 160 0.5 * 0.03f + // decline ratio * weight | 160 0.5 * 0.03f + // decline ratio * weight |
| 161 0.0 * 0.00f + // ignore ratio * (default) weight | 161 0.0 * 0.00f + // ignore ratio * (default) weight |
| 162 1.0 * 0.04f + // one-hot src-language "en" * weight | 162 1.0 * 0.04f + // one-hot src-language "en" * weight |
| 163 1.0 * 0.00f + // one-hot dst-language "fr" * weight | 163 1.0 * 0.00f + // one-hot dst-language "fr" * weight |
| 164 1.0 * 0.07f + // one-hot country * weight | 164 1.0 * 0.07f + // one-hot country * weight |
| 165 1.0 * 0.12f + // one-hot locale * weight | 165 1.0 * 0.12f + // one-hot locale * weight |
| 166 0.01f); // bias | 166 0.01f); // bias |
| 167 | 167 |
| 168 EXPECT_NEAR(expected, | 168 EXPECT_NEAR(expected, |
| 169 ranker->CalculateScore(0.5, 0.5, 0.0, "en", "fr", "zh-CN", "us"), | 169 ranker->CalculateScore(ranker->GetSharedModelPtr()->data, 0.5, |
| 170 0.5, 0.0, "en", "fr", "zh-CN", "us"), |
| 170 0.000001); | 171 0.000001); |
| 171 } | 172 } |
| 172 | 173 |
| 173 TEST_F(TranslateRankerTest, ShouldOfferTranslation) { | 174 TEST_F(TranslateRankerTest, ShouldOfferTranslation) { |
| 174 InitFeatures({kTranslateRankerQuery, kTranslateRankerEnforcement}, {}); | 175 InitFeatures({kTranslateRankerQuery, kTranslateRankerEnforcement}, {}); |
| 175 // With a bias of -0.5 en->fr is not over the threshold. | 176 // With a bias of -0.5 en->fr is not over the threshold. |
| 176 EXPECT_FALSE(GetRankerForTest(-0.5f)->ShouldOfferTranslation( | 177 EXPECT_FALSE(GetRankerForTest(-0.5f)->ShouldOfferTranslation( |
| 177 *translate_prefs_, "en", "fr")); | 178 *translate_prefs_, "en", "fr")); |
| 178 // With a bias of 0.25 en-fr is over the threshold. | 179 // With a bias of 0.25 en-fr is over the threshold. |
| 179 EXPECT_TRUE(GetRankerForTest(0.25f)->ShouldOfferTranslation(*translate_prefs_, | 180 EXPECT_TRUE(GetRankerForTest(0.25f)->ShouldOfferTranslation(*translate_prefs_, |
| (...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 222 ranker->RecordTranslateEvent(event_1); | 223 ranker->RecordTranslateEvent(event_1); |
| 223 ranker->RecordTranslateEvent(event_2); | 224 ranker->RecordTranslateEvent(event_2); |
| 224 ranker->RecordTranslateEvent(event_3); | 225 ranker->RecordTranslateEvent(event_3); |
| 225 | 226 |
| 226 // Logging is disabled, so no events should be cached. | 227 // Logging is disabled, so no events should be cached. |
| 227 ranker->FlushTranslateEvents(&flushed_events); | 228 ranker->FlushTranslateEvents(&flushed_events); |
| 228 EXPECT_EQ(0U, flushed_events.size()); | 229 EXPECT_EQ(0U, flushed_events.size()); |
| 229 } | 230 } |
| 230 | 231 |
| 231 } // namespace translate | 232 } // namespace translate |
| OLD | NEW |