1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/segmentation_platform/embedder/default_model/low_user_engagement_model.h"
#include <array>
#include "base/metrics/field_trial_params.h"
#include "base/task/sequenced_task_runner.h"
#include "components/segmentation_platform/internal/metadata/metadata_writer.h"
#include "components/segmentation_platform/public/config.h"
#include "components/segmentation_platform/public/constants.h"
#include "components/segmentation_platform/public/features.h"
namespace segmentation_platform {
namespace {
using proto::SegmentId;
// Default parameters for Chrome Start model.
constexpr SegmentId kChromeStartSegmentId =
SegmentId::OPTIMIZATION_TARGET_SEGMENTATION_CHROME_LOW_USER_ENGAGEMENT;
constexpr int64_t kChromeStartSignalStorageLength = 28;
constexpr int64_t kChromeStartMinSignalCollectionLength = 28;
constexpr int64_t kModelVersion = 2;
// InputFeatures.
constexpr std::array<MetadataWriter::UMAFeature, 1> kChromeStartUMAFeatures = {
MetadataWriter::UMAFeature{
.signal_type = proto::SignalType::HISTOGRAM_VALUE,
.name = "Session.TotalDuration",
.bucket_count = 28,
.tensor_length = 28,
.aggregation = proto::Aggregation::BUCKETED_COUNT,
.enum_ids_size = 0}};
} // namespace
LowUserEngagementModel::LowUserEngagementModel()
: DefaultModelProvider(kChromeStartSegmentId) {}
std::unique_ptr<Config> LowUserEngagementModel::GetConfig() {
if (!base::FeatureList::IsEnabled(
features::kSegmentationPlatformLowEngagementFeature)) {
return nullptr;
}
auto config = std::make_unique<Config>();
config->segmentation_key = kChromeLowUserEngagementSegmentationKey;
config->segmentation_uma_name = kChromeLowUserEngagementUmaName;
config->AddSegmentId(kChromeStartSegmentId,
std::make_unique<LowUserEngagementModel>());
config->auto_execute_and_cache = true;
config->is_boolean_segment = true;
return config;
}
std::unique_ptr<DefaultModelProvider::ModelConfig>
LowUserEngagementModel::GetModelConfig() {
proto::SegmentationModelMetadata chrome_start_metadata;
MetadataWriter writer(&chrome_start_metadata);
writer.SetDefaultSegmentationMetadataConfig(
kChromeStartMinSignalCollectionLength, kChromeStartSignalStorageLength);
// Set features.
writer.AddUmaFeatures(kChromeStartUMAFeatures.data(),
kChromeStartUMAFeatures.size());
// Set OutputConfig.
writer.AddOutputConfigForBinaryClassifier(
/*threshold=*/0.5f,
/*positive_label=*/kChromeLowUserEngagementUmaName,
/*negative_label=*/kLegacyNegativeLabel);
writer.AddPredictedResultTTLInOutputConfig(
/*top_label_to_ttl_list=*/{}, /*default_ttl=*/7,
/*time_unit=*/proto::TimeUnit::DAY);
return std::make_unique<ModelConfig>(std::move(chrome_start_metadata),
kModelVersion);
}
void LowUserEngagementModel::ExecuteModelWithInput(
const ModelProvider::Request& inputs,
ExecutionCallback callback) {
// Invalid inputs.
if (inputs.size() != 28) {
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE, base::BindOnce(std::move(callback), std::nullopt));
return;
}
float result = 0;
std::array<bool, 4> weeks = {};
for (unsigned i = 0; i < 28; ++i) {
int week_idx = i / 7;
weeks[week_idx] = weeks[week_idx] || inputs[i];
}
if (!weeks[0] || !weeks[1] || !weeks[2] || !weeks[3])
result = 1;
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE,
base::BindOnce(std::move(callback), ModelProvider::Response(1, result)));
}
} // namespace segmentation_platform
|
