Text Classification - TensorFlow - Amazon SageMaker AI

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Text Classification - TensorFlow

El algoritmo Text Classification - TensorFlow en Amazon SageMaker AI es un algoritmo de aprendizaje supervisado que admite el aprendizaje por transferencia con muchos modelos previamente entrenados desde TensorFlow Hub. Utilice el aprendizaje por transferencia para ajustar uno de los modelos previamente entrenados disponibles en su propio conjunto de datos, aunque no haya una gran cantidad de datos de texto disponibles. El algoritmo de clasificación de texto toma una cadena de texto como entrada y genera una probabilidad para cada una de las etiquetas de clase. Los conjuntos de datos de entrenamiento deben estar en formato CSV. Esta página incluye información sobre las recomendaciones de instancias de Amazon EC2 y cuadernos de muestra para Text Classification - TensorFlow.

Recomendaciones de instancias de Amazon EC2 para el algoritmo Text Classification - TensorFlow

El algoritmo Text Classification - TensorFlow es compatible con todas las instancias de CPU y GPU para el entrenamiento, lo que incluye lo siguiente:

  • ml.p2.xlarge

  • ml.p2.16xlarge

  • ml.p3.2xlarge

  • ml.p3.16xlarge

  • ml.g4dn.xlarge

  • ml.g4dn.16.xlarge

  • ml.g5.xlarge

  • ml.g5.48xlarge

Para el entrenamiento con lotes grandes, recomendamos utilizar instancias de GPU con más memoria. Para la inferencia, se pueden usar instancias de CPU (como M5) o de GPU (P2, P3, G4dn o G5). Para obtener una lista completa de las instancias de entrenamiento e inferencia en todas las regiones de AWS, consulte Precios de Amazon SageMaker.

Cuadernos de ejemplo de Text Classification - TensorFlow

Para obtener más información sobre cómo utilizar el algoritmo Text Classification - TensorFlow en SageMaker AI para el aprendizaje por transferencia en un conjunto de datos personalizado, consulte el cuaderno Introduction to JumpStart - Text Classification.

Para obtener instrucciones sobre cómo crear instancias de cuadernos de Jupyter que pueda utilizar para ejecutar el ejemplo en SageMaker AI, consulte Instancias de Amazon SageMaker Notebook. Después de crear y abrir una instancia de cuaderno, seleccione la pestaña Ejemplos de SageMaker AI para ver una lista de todas las muestras de SageMaker AI. Para abrir un cuaderno, elija su pestaña Usar y elija Crear copia.