TextSearcher

classe finale publique TextSearcher

Effectue une recherche de similarité sur une chaîne de texte.

L’API attend un modèle TFLite avec des métadonnées de modèle TFLite facultatives, mais fortement recommandées. .

L'API attend un modèle TFLite avec des métadonnées renseignées. Les métadonnées doivent contenir les informations suivantes :

  • Pour le modèle TFLite basé sur Bert :
    • 3 tenseurs d'entrée de type kTfLiteString avec les noms "ids", "mask" et "segment_ids".
    • input_process_units pour Wordpiece/Sentencepiece Tokenizer
    • exactement un tenseur de sortie de type kTfLiteFloat32
  • Pour le modèle TFLite basé sur Regex :
    • 1 tenseur d'entrée.
    • input_process_units pour RegexTokenizer Tokenizer
    • exactement un tenseur de sortie de type kTfLiteFloat32
  • Pour le modèle TFLite basé sur Universal Sentence Encoder :
    • 3 tenseurs d'entrée avec les noms "inp_text", "res_context" et "res_text"
    • 2 tenseurs de sortie avec les noms "query_encoding" et "response_encoding" de type kTfLiteFloat32

TODO(b/180502532) : ajouter un pointeur vers un exemple de modèle.

TODO(b/222671076) : ajoutez des méthodes de création d'usine sans options, telles que `createFromFile`, une fois que le format de fichier unique (fichier d'index emballé dans le modèle) est pris en charge.

Classes imbriquées

classe TextSearcher.TextSearcherOptions Options de configuration d'un TextSearcher.

Méthodes publiques

Recherche de texte statique
createFromBufferAndOptions (options ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )
Crée une instance TextSearcher avec un tampon de modèle et TextSearcher.TextSearcherOptions .
Recherche de texte statique
createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, options TextSearcher.TextSearcherOptions , int indexFd)
Recherche de texte statique
createFromFileAndOptions (options de fichier modelFile, TextSearcher.TextSearcherOptions )
Crée une instance TextSearcher .
Recherche de texte statique
createFromFileAndOptions (contexte contextuel, chaîne modelPath, options TextSearcher.TextSearcherOptions )
Crée une instance TextSearcher à partir de TextSearcher.TextSearcherOptions .
Liste < Voisin le plus proche >
recherche (texte de chaîne )
Effectue une extraction d'intégration sur l'entrée de chaîne fournie, suivie d'une recherche du voisin le plus proche dans l'index.

Méthodes héritées

Méthodes publiques

public static TextSearcher createFromBufferAndOptions (options ByteBuffer modelBuffer, TextSearcher.TextSearcherOptions )

Crée une instance TextSearcher avec un tampon de modèle et TextSearcher.TextSearcherOptions .

Paramètres
modèleBuffer un ByteBuffer direct ou un MappedByteBuffer du modèle de recherche
choix
Jetés
Exception d'argument illégal si le tampon de modèle n'est pas un ByteBuffer direct ou un MappedByteBuffer
IOException si une erreur d'E/S se produit lors du chargement du fichier d'index
IllegalStateException s'il y a une erreur interne
Exception d'exécution s'il y a une erreur autrement non spécifiée

public static TextSearcher createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, options TextSearcher.TextSearcherOptions , int indexFd)

Paramètres
modèleBuffer
choix
indexFd

public static TextSearcher createFromFileAndOptions (options de fichier modelFile, TextSearcher.TextSearcherOptions )

Crée une instance TextSearcher .

Paramètres
fichier modèle l'instance File du modèle de recherche
choix
Jetés
IOException si une erreur d'E/S se produit lors du chargement du modèle tflite ou du fichier d'index
Exception d'argument illégal si un argument n'est pas valide
IllegalStateException s'il y a une erreur interne
Exception d'exécution s'il y a une erreur autrement non spécifiée

public static TextSearcher createFromFileAndOptions (contexte contextuel, chaîne modelPath, options TextSearcher.TextSearcherOptions )

Crée une instance TextSearcher à partir de TextSearcher.TextSearcherOptions .

Paramètres
contexte
modèleChemin chemin du modèle de recherche avec métadonnées dans les ressources
choix
Jetés
IOException si une erreur d'E/S se produit lors du chargement du modèle tflite ou du fichier d'index
Exception d'argument illégal si un argument n'est pas valide
IllegalStateException s'il y a une erreur interne
Exception d'exécution s'il y a une erreur autrement non spécifiée

liste publique < NearestNeighbour > recherche (texte de chaîne )

Effectue une extraction d'intégration sur l'entrée de chaîne fournie, suivie d'une recherche du voisin le plus proche dans l'index.

Paramètres
texte saisir une requête de texte dans le modèle