BertNLClassifier

classe publique BertNLClassifier

API Classifier pour les tâches NLClassification avec les modèles Bert, catégorise la chaîne en différentes classes. L'API attend un modèle TFLite basé sur Bert avec des métadonnées renseignées.

Les métadonnées doivent contenir les informations suivantes :

  • 1 input_process_unit pour Wordpiece/Sentencepiece Tokenizer.
  • 3 tenseurs d'entrée avec les noms "ids", "mask" et "segment_ids".
  • 1 tenseur de sortie de type float32[1, 2], avec un fichier d'étiquette éventuellement joint. Si un fichier d'étiquettes est joint, le fichier doit être un fichier texte brut avec une étiquette par ligne, le nombre d'étiquettes doit correspondre au nombre de catégories générées par le modèle.

Classes imbriquées

classe BertNLClassifier.BertNLClassifierOptions Options pour configurer BertNLClassifier.

Méthodes publiques

Liste < Catégorie >
classer (texte de chaîne )
Effectue une classification sur une entrée de chaîne, renvoie des Category classifiées.
BertNLClassifier statique
createFromBuffer ( ByteBuffer modelBuffer)
Crée BertNLClassifier avec un tampon de modèle et BertNLClassifier.BertNLClassifierOptions par défaut.
BertNLClassifier statique
createFromBufferAndOptions (options ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )
Crée BertNLClassifier avec un tampon de modèle et BertNLClassifier.BertNLClassifierOptions .
BertNLClassifier statique
createFromFile (contexte contextuel, String modelPath)
Crée BertNLClassifier à partir d'un fichier modèle avec des métadonnées et BertNLClassifier.BertNLClassifierOptions par défaut.
BertNLClassifier statique
createFromFile ( Fichier modèleFichier)
Crée BertNLClassifier à partir d'un objet File avec des métadonnées et BertNLClassifier.BertNLClassifierOptions par défaut.
BertNLClassifier statique
createFromFileAndOptions (options de fichier modelFile, BertNLClassifier.BertNLClassifierOptions )
Crée BertNLClassifier à partir d'un objet File avec des métadonnées et BertNLClassifier.BertNLClassifierOptions .
BertNLClassifier statique
createFromFileAndOptions (contexte contextuel, chaîne modelPath, options BertNLClassifier.BertNLClassifierOptions )
Crée BertNLClassifier à partir d'un fichier modèle avec des métadonnées et BertNLClassifier.BertNLClassifierOptions .

Méthodes héritées

Méthodes publiques

liste publique < Catégorie > classifier (Texte de chaîne )

Effectue une classification sur une entrée de chaîne, renvoie des Category classifiées.

Paramètres
texte saisir du texte dans le modèle.
Retour
  • Une liste des résultats de catégorie.

public statique BertNLClassifier createFromBuffer ( ByteBuffer modelBuffer)

Crée BertNLClassifier avec un tampon de modèle et BertNLClassifier.BertNLClassifierOptions par défaut.

Paramètres
modèleBuffer un ByteBuffer direct ou un MappedByteBuffer du modèle
Retour
Jetés
Exception d'argument illégal si le tampon de modèle n'est pas un ByteBuffer direct ou un MappedByteBuffer
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 BertNLClassifier createFromBufferAndOptions (options ByteBuffer modelBuffer, BertNLClassifier.BertNLClassifierOptions )

Crée BertNLClassifier avec un tampon de modèle et BertNLClassifier.BertNLClassifierOptions .

Paramètres
modèleBuffer un ByteBuffer direct ou un MappedByteBuffer du modèle
choix pour configurer le classificateur
Retour
Jetés
Exception d'argument illégal si le tampon de modèle n'est pas un ByteBuffer direct ou un MappedByteBuffer
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 BertNLClassifier createFromFile (contexte contextuel, String modelPath)

Crée BertNLClassifier à partir d'un fichier modèle avec des métadonnées et BertNLClassifier.BertNLClassifierOptions par défaut.

Paramètres
contexte Contexte Android
modèleChemin Chemin d'accès au modèle de classification
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 BertNLClassifier createFromFile ( Fichier modelFile)

Crée BertNLClassifier à partir d'un objet File avec des métadonnées et BertNLClassifier.BertNLClassifierOptions par défaut.

Paramètres
fichier modèle Le modèle de classification Instance File
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 BertNLClassifier createFromFileAndOptions (options de fichier modelFile, BertNLClassifier.BertNLClassifierOptions )

Crée BertNLClassifier à partir d'un objet File avec des métadonnées et BertNLClassifier.BertNLClassifierOptions .

Paramètres
fichier modèle Le modèle de classification Instance File
choix pour configurer le classificateur
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 BertNLClassifier createFromFileAndOptions (contexte contextuel, String modelPath, options BertNLClassifier.BertNLClassifierOptions )

Crée BertNLClassifier à partir d'un fichier modèle avec des métadonnées et BertNLClassifier.BertNLClassifierOptions .

Paramètres
contexte Contexte Android.
modèleChemin Chemin d'accès au modèle de classification
choix pour configurer le classificateur
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