Vous êtes sur la page 1sur 2

Voici un exemple de code pour une application similaire à une IA en utilisant le langage

Python et la bibliothèque TensorFlow par HarDi Sukami :

```python
import tensorflow as tf

# Définition du modèle
input_size = 784
output_size = 10
hidden_layer_size = 50

model = tf.keras.Sequential([
tf.keras.layers.Dense(hidden_layer_size, activation='relu'),
tf.keras.layers.Dense(hidden_layer_size, activation='relu'),
tf.keras.layers.Dense(output_size, activation='softmax')
])

# Chargement des données d'entraînement


mnist = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()

# Prétraitement des données


x_train = x_train.reshape(-1, input_size) / 255.0
x_test = x_test.reshape(-1, input_size) / 255.0

# Entraînement du modèle
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])

batch_size = 100
max_epochs = 10

model.fit(x_train, y_train,
batch_size=batch_size,
epochs=max_epochs,
validation_data=(x_test, y_test))

# Évaluation du modèle
model.evaluate(x_test, y_test)

# Utilisation du modèle pour prédire de nouvelles données


predictions = model.predict(x_test[:5])
print(predictions)
```

Ce code utilise la bibliothèque TensorFlow pour créer un modèle de réseau de neurones


artificiels avec deux couches cachées et une couche de sortie. Le modèle est entraîné sur le
jeu de données MNIST, qui contient des images de chiffres écrits à la main, et utilisé pour
prédire les chiffres dans le jeu de test.

Notez que ce code simplifié peut être optimisé et adapté en fonction des besoins spécifiques
de votre application.

Vous aimerez peut-être aussi