chromium/components/assist_ranker/base_predictor.cc

// Copyright 2017 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/assist_ranker/base_predictor.h"

#include "base/containers/contains.h"
#include "base/feature_list.h"
#include "components/assist_ranker/proto/ranker_example.pb.h"
#include "components/assist_ranker/proto/ranker_model.pb.h"
#include "components/assist_ranker/ranker_example_util.h"
#include "components/assist_ranker/ranker_model.h"
#include "services/metrics/public/cpp/ukm_entry_builder.h"
#include "services/metrics/public/cpp/ukm_recorder.h"
#include "url/gurl.h"

namespace assist_ranker {

BasePredictor::BasePredictor(const PredictorConfig& config) : … { … }

BasePredictor::~BasePredictor() { … }

void BasePredictor::LoadModel(std::unique_ptr<RankerModelLoader> model_loader) { … }

void BasePredictor::OnModelAvailable(
    std::unique_ptr<assist_ranker::RankerModel> model) { … }

bool BasePredictor::IsReady() { … }

void BasePredictor::LogFeatureToUkm(const std::string& feature_name,
                                    const Feature& feature,
                                    ukm::UkmEntryBuilder* ukm_builder) { … }

void BasePredictor::LogExampleToUkm(const RankerExample& example,
                                    ukm::SourceId source_id) { … }

std::string BasePredictor::GetModelName() const { … }

GURL BasePredictor::GetModelUrl() const { … }

float BasePredictor::GetPredictThresholdReplacement() const { … }

RankerExample BasePredictor::PreprocessExample(const RankerExample& example) { … }

}  // namespace assist_ranker