chromium/components/assist_ranker/generic_logistic_regression_inference.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/generic_logistic_regression_inference.h"

#include <cmath>

#include "base/logging.h"
#include "components/assist_ranker/example_preprocessing.h"
#include "components/assist_ranker/ranker_example_util.h"

namespace assist_ranker {

const float kDefaultThreshold = …;

float Sigmoid(float x) { … }

GenericLogisticRegressionInference::GenericLogisticRegressionInference(
    GenericLogisticRegressionModel model_proto)
    : … { … }

float GenericLogisticRegressionInference::PredictScore(
    const RankerExample& example) { … }

bool GenericLogisticRegressionInference::Predict(const RankerExample& example) { … }

float GenericLogisticRegressionInference::GetThreshold() { … }

}  // namespace assist_ranker