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
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
- une instance
BertNLClassifier
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
- une instance
BertNLClassifier
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
- une instance
BertNLClassifier
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
- une instance
BertNLClassifier
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
- une instance
BertNLClassifier
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
- une instance
BertNLClassifier
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 |