Modèles multi-sorties
L’API Functional de Keras permet de créer des modèles complexes avec plusieurs sorties. Cela est particulièrement utile pour les tâches de multi-tâches où le modèle doit produire plusieurs prédictions simultanément. Voici comment vous pouvez utiliser l’API Functional pour créer un modèle qui produit plusieurs sorties.
Étape 1 : Importer les modules nécessaires
Commencez par importer les modules nécessaires de TensorFlow et Keras.
Étape 2 : Définir les entrées
Utilisez la classe Input
pour définir les entrées du modèle. Vous pouvez spécifier la forme des données d’entrée.
Étape 3 : Définir les couches partagées
Ajoutez des couches partagées qui seront utilisées pour produire les différentes sorties.
Étape 4 : Définir les sorties
Ajoutez des couches de sortie pour chaque tâche. Chaque sortie peut avoir sa propre architecture.
Étape 5 : Définir le modèle
Utilisez la classe Model
pour créer le modèle en spécifiant les entrées et les sorties.
Étape 6 : Compiler le modèle
Compilez le modèle en spécifiant l’optimiseur, les fonctions de perte pour chaque sortie, et les métriques.
Étape 7 : Afficher le résumé du modèle
Affichez un résumé de l’architecture du modèle pour vérifier sa structure.
Exemple complet
Voici un exemple complet de la création d’un modèle qui produit plusieurs sorties en utilisant l’API Functional :
Explication détaillée
- Importer les modules nécessaires :
- Importez les modules nécessaires de TensorFlow et Keras.
- Définir les entrées :
- Utilisez la classe
Input
pour définir la forme des données d’entrée. Dans cet exemple, l’entrée a une forme de(32,)
.
- Utilisez la classe
- Définir les couches partagées :
- Ajoutez des couches partagées qui seront utilisées pour produire les différentes sorties. Dans cet exemple, deux couches denses avec 64 unités et activation ReLU sont ajoutées.
- Définir les sorties :
- Ajoutez des couches de sortie pour chaque tâche. Chaque sortie peut avoir sa propre architecture. Dans cet exemple, deux couches de sortie avec 1 unité et activation sigmoid sont ajoutées.
- Définir le modèle :
- Utilisez la classe
Model
pour créer le modèle en spécifiant les entrées et les sorties.
- Utilisez la classe
- Compiler le modèle :
- Compilez le modèle avec l’optimiseur Adam, les fonctions de perte
binary_crossentropy
pour chaque sortie, et les métriques d’exactitude.
- Compilez le modèle avec l’optimiseur Adam, les fonctions de perte
- Afficher le résumé du modèle :
- Utilisez la méthode
summary
pour afficher un résumé de l’architecture du modèle.
- Utilisez la méthode
En utilisant l’API Functional, vous pouvez créer des modèles complexes avec plusieurs sorties, adaptés à des tâches de multi-tâches et à des architectures avancées.