blob: 43706a08dd09635c337009d2c8555f14a35451cc [file] [log] [blame]
// Copyright (c) 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
syntax = "proto2";
option optimize_for = LITE_RUNTIME;
import "example_preprocessor.proto";
package assist_ranker;
// A vector of floating-point values.
message FloatVector {
repeated float values = 1;
// The weights and biases for a single neural-network layer.
message NNLayer {
// The weights for the layer.
repeated FloatVector weights = 1;
// The bias vectors for the layer.
optional FloatVector biases = 2;
// Defines the model weights and biases for a neural network with a single
// hidden layer.
message NNClassifierModel {
// The single hidden layer.
optional NNLayer hidden_layer = 1;
// The output logits layer.
optional NNLayer logits_layer = 2;
// The preprocessing config for this model. This is used to vectorize the
// examples so they can be used as input for the classifier.
optional ExamplePreprocessorConfig preprocessor_config = 3;