BertQuestionAnswerer

classe publique BertQuestionAnswerer

Renvoie le plus de réponses possibles à une question donnée pour les modèles d'assurance qualité (BERT, Albert, etc.).

L'API attend un modèle TFLite basé sur Bert avec des métadonnées contenant les informations suivantes :

  • input_process_units pour Wordpiece/Sentencepiece Tokenizer - Wordpiece Tokenizer peut être utilisé pour un modèle MobileBert , Sentencepiece Tokenizer Tokenizer peut être utilisé pour un modèle Albert .
  • 3 tenseurs d'entrée avec les noms "ids", "mask" et "segment_ids".
  • 2 tenseurs de sortie avec les noms "end_logits" et "start_logits".

Classes imbriquées

classe BertQuestionAnswerer.BertQuestionAnswererOptions Options de configuration d'un BertQuestionAnswerer .

Méthodes publiques

Liste < QaAnswer >
réponse (contexte de chaîne , question de chaîne )
Répond à la question en fonction du contexte et renvoie une liste de QaAnswer possibles.
BertQuestionAnswerer statique
createAlbertQuestionAnswererFromFile (contexte contextuel, chaîne modelPath, chaîne phrasePièceModelPath)
Crée une instance BertQuestionAnswerer avec un modèle Albert et un fichier modèle de morceau de phrase.
BertQuestionAnswerer statique
createBertQuestionAnswererFromFile (contexte contextuel, String modelPath, String vocabPath)
Crée une instance BertQuestionAnswerer avec un modèle Bert et un fichier de vocabulaire.
BertQuestionAnswerer statique
createFromFile (contexte contextuel, String modelPath)
Crée une instance BertQuestionAnswerer à partir de la valeur par défaut BertQuestionAnswerer.BertQuestionAnswererOptions .
BertQuestionAnswerer statique
createFromFile ( Fichier modèleFichier)
Crée une instance BertQuestionAnswerer à partir de la valeur par défaut BertQuestionAnswerer.BertQuestionAnswererOptions .
BertQuestionAnswerer statique
BertQuestionAnswerer statique

Méthodes héritées

Méthodes publiques

liste publique < QaAnswer > réponse (contexte de chaîne , question de chaîne )

Répond à la question en fonction du contexte et renvoie une liste de QaAnswer possibles. Peut être vide si aucune réponse n'est trouvée dans le contexte donné.

Paramètres
contexte contexte sur lequel la question se base
question question à poser
Retour
  • une liste de réponses possibles dans QaAnswer

public static BertQuestionAnswerer createAlbertQuestionAnswererFromFile (contexte contextuel, String modelPath, String sentencePièceModelPath)

Crée une instance BertQuestionAnswerer avec un modèle Albert et un fichier modèle de morceau de phrase.

Un modèle approprié est : https://tfhub.dev/tensorflow/lite-model/albert_lite_base/squadv1/1

Paramètres
contexte contexte Android
modèleChemin chemin de fichier vers le modèle Albert. Remarque : Le modèle ne doit pas être compressé
phrasePièceModèlePath chemin d'accès au fichier modèle de morceau de phrase. Remarque : Le modèle ne doit pas être compressé
Retour
Jetés
IOException Si le fichier modèle ne parvient pas à se charger
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 BertQuestionAnswerer createBertQuestionAnswererFromFile (contexte contextuel, String modelPath, String vocabPath)

Crée une instance BertQuestionAnswerer avec un modèle Bert et un fichier de vocabulaire.

Un modèle approprié est : https://tfhub.dev/tensorflow/lite-model/mobilebert/1/default/1

Paramètres
contexte contexte Android
modèleChemin chemin du fichier vers le modèle Bert. Remarque : Le modèle ne doit pas être compressé
vocabulairePath chemin d'accès au fichier de vocabulaire. Remarque : le fichier ne doit pas être compressé
Retour
Jetés
IOException Si le fichier modèle ne parvient pas à se charger
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 BertQuestionAnswerer createFromFile (contexte contextuel, String modelPath)

Crée une instance BertQuestionAnswerer à partir de la valeur par défaut BertQuestionAnswerer.BertQuestionAnswererOptions .

Paramètres
contexte contexte Android
modèleChemin chemin du fichier vers le modèle avec les métadonnées. Remarque : Le modèle ne doit pas être compressé
Retour
Jetés
IOException si le fichier modèle ne parvient pas à se charger
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 statique BertQuestionAnswerer createFromFile ( Fichier modelFile)

Crée une instance BertQuestionAnswerer à partir de la valeur par défaut BertQuestionAnswerer.BertQuestionAnswererOptions .

Paramètres
fichier modèle un objet File du modèle
Retour
Jetés
IOException si le fichier modèle ne parvient pas à se charger
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 BertQuestionAnswerer createFromFileAndOptions (options de fichier modelFile, BertQuestionAnswerer.BertQuestionAnswererOptions )

Paramètres
fichier modèle un objet File du modèle
choix
Retour
Jetés
IOException si le fichier modèle ne parvient pas à se charger
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 BertQuestionAnswerer createFromFileAndOptions (contexte contextuel, String modelPath, options BertQuestionAnswerer.BertQuestionAnswererOptions )

Paramètres
contexte contexte Android
modèleChemin chemin du fichier vers le modèle avec les métadonnées. Remarque : Le modèle ne doit pas être compressé
choix
Retour
Jetés
IOException si le fichier modèle ne parvient pas à se charger
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