123456789101112131415161718192021222324252627282930313233343536373839 |
- // 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;
- }
|