Réseaux de neurones avec Tensorflow

Voici un exemple de petit modèle de réseaux de neurones qui apprend à faire une prédiction à partir d'un nombre donné en entrée.

Extrait de code utilisé :

from tensorflow import keras
from tensorflow.keras import layers
model = keras.Sequential()
model.add(layers.Dense(units=6, input_shape=[1])) model.add(layers.Dense(units=128)) model.add(layers.Dense(units=128)) model.add(layers.Dense(units=128)) model.add(layers.Dense(units=128)) model.add(layers.Dense(units=128))
model.add(layers.Dense(units=1))

entree = [10, 24, 31, 40, 55]
sortie = [48, 90, 111, 138, 183]

model.compile(loss='mean_squared_error', optimizer='Adam') model.fit(x=entree, y=sortie, epochs=1000)
while True: x = int(input('Nombre :')) print('Prédiction :'+str(model.predict([x])))

Exemple de resultat

Si on donne le nombre 75 en entrée, le modèle prédit une sortie de 242.99997, le résultat attendu étant de 243. Je vous laisse trouver le calcul !

Ce modèle est simple, mais il montre ma capacité à entraîner un réseau de neurones pour effectuer des prédictions.

Explications pédagogiques :

Retour aux services