Classification textuelle : TensorFlow

L’algorithme Text Classification - TensorFlow dans Amazon SageMaker AI est un algorithme d’apprentissage supervisé qui prend en charge l’apprentissage par transfert avec de nombreux modèles pré-entraînés provenant du hub TensorFlow. Utilisez l'apprentissage par transfert pour affiner l'un des modèles pré-entraînés disponibles sur votre propre jeu de données, même si une grande quantité de données de texte n'est pas disponible. L'algorithme de classification de texte prend une image en entrée et génère en sortie une probabilité pour chaque étiquette de classe fournie. Les jeux de données de formation doivent être au format CSV. Cette page contient des informations sur les recommandations relatives aux instances Amazon EC2 et des exemples de blocs-notes pour Text Classification - TensorFlow.

Recommandation d'instance Amazon EC2 pour l'algorithme Text Classification - TensorFlow

L'algorithme Text Classification - TensorFlow prend en charge toutes les instances de CPU et de GPU pour l'entraînement, y compris :

Nous recommandons d'utiliser les instances de GPU avec davantage de mémoire pour l'entraînement avec de grandes tailles de lot. Les instances de CPU (telles que M5) et de GPU (P2, P3, G4dn ou G5) peuvent être utilisées pour l'inférence. Pour obtenir une liste complète des instances de formation et d'inférence de SageMaker dans toutes les régions AWS, consultez la Tarification d'Amazon SageMaker.

Exemples de blocs-notes Text Classification - TensorFlow

Pour plus d’informations sur l’utilisation de l’algorithme Text Classification – TensorFlow SageMaker AI pour l’apprentissage par transfert sur un jeu de données personnalisé, consultez le bloc-notes Présentation de JumpStart – Classification de texte.

Pour savoir comment créer des instances de blocs-notes Jupyter et y accéder afin de les utiliser pour exécuter l’exemple dans SageMaker AI, consultez Instances de bloc-notes Amazon SageMaker. Après avoir créé et ouvert une instance de bloc-notes, sélectionnez l’onglet Exemples SageMaker AI pour afficher la liste de tous les exemples SageMaker AI. Pour ouvrir un bloc-notes, choisissez son onglet Use (Utiliser), puis Create copy (Créer une copie).