La tarea MediaPipe Text Classifier te permite clasificar texto en un conjunto de categorías definidas, como opiniones positivas o negativas. Las categorías se determinan según el modelo que usas y cómo se entrenó ese modelo. Estas instrucciones te muestran cómo usar el clasificador de texto en apps web y de JavaScript.
Para ver esta tarea en acción, consulta el demostración. Para obtener más información sobre las funciones, los modelos y de configuración de esta tarea, consulta la Descripción general.
Ejemplo de código
El código de ejemplo para el clasificador de texto proporciona una implementación completa de este en JavaScript para tu referencia. Este código te ayuda a probar la tarea y comienza a compilar tu propia app de clasificación de texto. Puedes ver, ejecutar y editar el código de ejemplo del clasificador de texto usando solo el navegador web.
Configuración
En esta sección, se describen los pasos clave para configurar tu entorno de desarrollo y proyectos de código específicamente para usar el clasificador de texto. Para obtener información general configurar tu entorno de desarrollo para usar tareas de MediaPipe, incluidas de la versión de la plataforma, consulta la Guía de configuración para la Web.
Paquetes de JavaScript
El código del clasificador de texto está disponible a través de la
@mediapipe/tasks-text
. Puedes encontrar y descargar estas bibliotecas en los vínculos que se proporcionan en la
plataforma
Guía de configuración.
Puedes instalar los paquetes obligatorios con el siguiente código para la etapa de pruebas local con el siguiente comando:
npm install @mediapipe/tasks-text
Si quieres realizar la implementación en un servidor, puedes usar un modelo de red de VPC (CDN), como jsDelivr a agrega el código directamente a tu página HTML de la siguiente manera:
<head>
<script src="https://cdn.jsdelivr.net/npm/@mediapipe/tasks-text@0.1/text-bundle.js"
crossorigin="anonymous"></script>
</head>
Modelo
La tarea del clasificador de texto MediaPipe requiere un modelo entrenado que sea compatible con esta tarea. Si deseas obtener más información sobre los modelos entrenados disponibles para el clasificador de texto, consulta la sección Modelos de la descripción general de la tarea.
Selecciona y descarga un modelo y, luego, guárdalo en el directorio de tu proyecto:
<dev-project-root>/assets/bert_text_classifier.tflite
Especifica la ruta del modelo con el objeto modelAssetPath de baseOptions
según se muestra a continuación:
baseOptions: {
modelAssetPath: `/assets/bert_text_classifier.tflite`
}
Crea la tarea
Usa una de las funciones TextClassifier.createFrom...() del clasificador de texto para lo siguiente:
preparan la tarea para ejecutar inferencias. Puedes usar el createFromModelPath()
con una ruta de acceso absoluta o relativa al archivo de modelo entrenado. El código
en el siguiente ejemplo se demuestra el uso de TextClassifier.createFromOptions()
. Para obtener más información sobre las opciones de configuración disponibles, consulta
Opciones de configuración.
En el siguiente código, se muestra cómo compilar y configurar esta tarea:
async function createClassifier() {
const textFiles = await FilesetResolver.forTextTasks("https://cdn.jsdelivr.net/npm/@mediapipe/tasks-text@latest/wasm/");
textClassifier = await TextClassifier.createFromOptions(
textFiles,
{
baseOptions: {
modelAssetPath: `https://storage.googleapis.com/mediapipe-tasks/text_classifier/bert_text_classifier.tflite`
},
maxResults: 5
}
);
}
createClassifier();
Opciones de configuración
Esta tarea tiene las siguientes opciones de configuración para la Web y JavaScript aplicaciones:
| Nombre de la opción | Descripción | Rango de valores | Valor predeterminado |
|---|---|---|---|
displayNamesLocale |
Configura el idioma de las etiquetas que se usarán para los nombres visibles que se proporcionan en la
metadatos del modelo de la tarea, si están disponibles. El valor predeterminado es en para
Inglés. Puedes agregar etiquetas localizadas a los metadatos de un modelo personalizado
con la API de Metadata Writer de TensorFlow Lite
| Código de configuración regional | en |
maxResults |
Establece el número máximo opcional de resultados de la clasificación con puntuación más alta en el resultado. Si < 0, se devolverán todos los resultados disponibles. | Cualquier número positivo | -1 |
scoreThreshold |
Establece el umbral de puntuación de predicción que anula el que se proporciona en los metadatos del modelo (si los hay). Se rechazarán los resultados por debajo de este valor. | Cualquier número de punto flotante | Sin establecer |
categoryAllowlist |
Establece la lista opcional de nombres de categorías permitidas. Si no está vacío,
los resultados de clasificación cuyo nombre de categoría no se encuentre en este conjunto serán
filtrado. Se ignoran los nombres de categoría duplicados o desconocidos.
Esta opción es mutuamente excluyente con categoryDenylist y usa
ambos darán como resultado un error. |
Cualquier cadena | Sin establecer |
categoryDenylist |
Establece la lista opcional de nombres de categorías que no están permitidos. Si
no vacío, los resultados de clasificación cuyo nombre de categoría se encuentre en este conjunto se filtrarán
y sale de ella. Se ignoran los nombres de categoría duplicados o desconocidos. Esta opción es mutuamente
excluyente con categoryAllowlist y usar ambos dará como resultado un error. |
Cualquier cadena | Sin establecer |
Preparar los datos
El clasificador de texto funciona con datos de texto (String). La tarea controla
Procesamiento previo de la entrada de datos, incluida la asignación de token y el procesamiento previo del tensor
Todo el procesamiento previo se controla dentro de la función classify(). No es necesario
para un procesamiento previo adicional del texto de entrada.
const inputText = "The input text to be classified.";
Ejecuta la tarea
El clasificador de texto usa la función classify() para activar inferencias. Para texto
clasificación, esto significa devolver las posibles categorías para el texto de entrada.
En el siguiente código, se muestra cómo ejecutar el procesamiento con la tarea un modelo de responsabilidad compartida.
// Wait to run the function until inner text is set
const result: TextClassifierResult = await textClassifier.classify(
inputText
);
Cómo controlar y mostrar resultados
El clasificador de texto genera una TextClassifierResult que contiene la lista.
de categorías posibles para el texto de entrada. Las categorías se definen según el
modelo que usas, así que si quieres diferentes categorías, elige un modelo diferente,
o volver a entrenar uno existente.
A continuación, se muestra un ejemplo de los datos de salida de esta tarea:
TextClassificationResult:
Classification #0 (single classification head):
ClassificationEntry #0:
Category #0:
category name: "positive"
score: 0.8904
index: 0
Category #1:
category name: "negative"
score: 0.1096
index: 1
Este resultado se obtuvo mediante la ejecución del clasificador de BERT en el texto de entrada:
"an imperfect but overall entertaining mystery"