chromium/components/segmentation_platform/embedder/default_model/metrics_clustering.cc

// Copyright 2024 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/metrics_clustering.h"

#include <memory>

#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/proto/aggregation.pb.h"
#include "components/segmentation_platform/public/proto/model_metadata.pb.h"

namespace segmentation_platform {
namespace features {
BASE_FEATURE(…);
}

namespace {
SegmentId;

// Default parameters for MetricsClustering model.
constexpr SegmentId kSegmentId = …;
constexpr int64_t kModelVersion = …;
constexpr int64_t kSignalStorageLength = …;
constexpr int64_t kMinSignalCollectionLength = …;
constexpr int64_t kResultTTLDays = …;

// InputFeatures.

// Set UMA metrics to use as input.
// TODO(b/355993452): Fill in the necessary signals for prediction.
constexpr std::array<MetadataWriter::UMAFeature, 0> kUMAFeatures = …;

}  // namespace

// static
std::unique_ptr<Config> MetricsClustering::GetConfig() { … }

MetricsClustering::MetricsClustering() : … { … }

std::unique_ptr<DefaultModelProvider::ModelConfig>
MetricsClustering::GetModelConfig() { … }

void MetricsClustering::ExecuteModelWithInput(
    const ModelProvider::Request& inputs,
    ExecutionCallback callback) { … }

}  // namespace segmentation_platform