chromium/components/autofill/core/browser/ml_model/autofill_model_encoder.h

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_AUTOFILL_CORE_BROWSER_ML_MODEL_AUTOFILL_MODEL_ENCODER_H_
#define COMPONENTS_AUTOFILL_CORE_BROWSER_ML_MODEL_AUTOFILL_MODEL_ENCODER_H_

#include <stdint.h>

#include <string>
#include <string_view>

#include "base/containers/flat_map.h"
#include "base/types/strong_alias.h"
#include "third_party/protobuf/src/google/protobuf/repeated_ptr_field.h"

namespace autofill {

class AutofillField;
class FormStructure;

// The Encoder performs vectorization for on-device Autofill field type
// prediction ML model. It changes the string input for preprocessing by
// standardizing and tokenizing it. Tokenization maps raw strings to tokens,
// and tokens to IDs based on the given dictionary. Empty Strings map to
// value 0 and unknown words map to value 1.
class AutofillModelEncoder { … };

}  // namespace autofill

#endif  // COMPONENTS_AUTOFILL_CORE_BROWSER_ML_MODEL_AUTOFILL_MODEL_ENCODER_H_