Maison > Périphériques technologiques > IA > le corps du texte

Créez des modèles d'apprentissage profond avec TensorFlow et Keras

王林
Libérer: 2024-01-24 09:18:05
avant
517 Les gens l'ont consulté

Créez des modèles dapprentissage profond avec TensorFlow et Keras

TensorFlow et Keras sont actuellement l'un des frameworks d'apprentissage en profondeur les plus populaires. Ils fournissent non seulement des API de haut niveau pour faciliter la création et la formation de modèles d'apprentissage profond, mais fournissent également une variété de couches et de types de modèles pour faciliter la construction de divers types de modèles d'apprentissage profond. Par conséquent, ils sont largement utilisés pour former des modèles d’apprentissage profond à grande échelle.

Nous utiliserons TensorFlow et Keras pour créer un modèle d'apprentissage profond pour la classification d'images. Dans cet exemple, nous utiliserons l'ensemble de données CIFAR-10, qui contient 10 catégories différentes avec 6 000 images couleur 32x32 par catégorie.

Tout d'abord, nous devons importer les bibliothèques et les ensembles de données nécessaires. Nous utiliserons TensorFlow version 2.0 et l'API Keras pour créer le modèle. Voici le code pour importer la bibliothèque et l'ensemble de données : ```python importer tensorflow astf à partir de tensorflow importer des keras à partir de tensorflow.keras.datasets importer mnist #Importer un ensemble de données (x_train, y_train), (x_test, y_test) = mnist.load_data() ``` Ce qui précède est le code pour importer la bibliothèque et l'ensemble de données. Nous utilisons la bibliothèque « tensorflow » pour construire le modèle et utilisons l'ensemble de données « mnist » comme exemple d'ensemble de données.

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras.datasets import cifar10

# 加载CIFAR-10数据集
(x_train, y_train), (x_test, y_test) = cifar10.load_data()

# 将像素值缩放到0到1之间
x_train = x_train.astype("float32") / 255.0
x_test = x_test.astype("float32") / 255.0

# 将标签从整数转换为one-hot编码
y_train = keras.utils.to_categorical(y_train, 10)
y_test = keras.utils.to_categorical(y_test, 10)
Copier après la connexion

Ensuite, nous définirons un modèle de réseau neuronal convolutif. Nous utiliserons trois couches convolutives et trois couches de pooling pour extraire les fonctionnalités, puis deux couches entièrement connectées pour la classification. Voici la définition de notre modèle :

model = keras.Sequential(
    [
        # 第一个卷积层
        layers.Conv2D(32, (3, 3), activation="relu", input_shape=(32, 32, 3)),
        layers.MaxPooling2D((2, 2)),
        # 第二个卷积层
        layers.Conv2D(64, (3, 3), activation="relu"),
        layers.MaxPooling2D((2, 2)),
        # 第三个卷积层
        layers.Conv2D(128, (3, 3), activation="relu"),
        layers.MaxPooling2D((2, 2)),
        # 展平层
        layers.Flatten(),
        # 全连接层
       layers.Dense(128, activation="relu"),
        layers.Dense(10, activation="softmax"),
    ]
)
Copier après la connexion

Dans ce modèle, nous utilisons la fonction d'activation ReLU, qui est une fonction non linéaire couramment utilisée qui peut aider le modèle à apprendre des relations non linéaires complexes. Nous avons également utilisé la fonction d'activation softmax pour la classification multi-classe.

Maintenant, nous pouvons compiler le modèle et commencer la formation. Nous utiliserons l'optimiseur Adam et la fonction de perte d'entropie croisée pour la formation du modèle. Voici le code : model.compile(optimizer='adam', loss='categorical_crossentropy') model.fit(X_train, y_train, epochs=10, batch_size=32)

# 编译模型
model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])

# 训练模型
history = model.fit(x_train, y_train, epochs=10, validation_data=(x_test, y_test))
Copier après la connexion

Une fois la formation terminée, nous pouvons utiliser l'ensemble de tests pour évaluer les performances du modèle. Voici notre code pour évaluer le modèle :

# 在测试集上评估模型
test_loss, test_acc = model.evaluate(x_test, y_test)

print("Test loss:", test_loss)
print("Test accuracy:", test_acc)
Copier après la connexion

Enfin, nous pouvons utiliser l'historique d'entraînement pour tracer la perte d'entraînement et de validation et la précision du modèle. Voici le code pour dessiner l'historique de la formation :

import matplotlib.pyplot as plt

# 绘制训练和验证损失
plt.plot(history.history["loss"], label="Training loss")
plt.plot(history.history["val_loss"], label="Validation loss")
plt.xlabel("Epoch")
plt.ylabel("Loss")
plt.legend()

plt.show()

# 绘制训练和验证准确率
plt.plot(history.history["accuracy"], label="Training accuracy")
plt.plot(history.history["val_accuracy"], label="Validation accuracy")
plt.xlabel("Epoch")
plt.ylabel("Accuracy")
plt.legend()

plt.show()
Copier après la connexion

Ce qui précède est le code complet d'un exemple de modèle d'apprentissage profond basé sur TensorFlow et Keras. Nous avons construit un modèle de réseau neuronal convolutif en utilisant l'ensemble de données CIFAR-10 pour les tâches de classification d'images.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Étiquettes associées:
source:163.com
Déclaration de ce site Web
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn
Tutoriels populaires
Plus>
Derniers téléchargements
Plus>
effets Web
Code source du site Web
Matériel du site Web
Modèle frontal
À propos de nous Clause de non-responsabilité Sitemap
Site Web PHP chinois:Formation PHP en ligne sur le bien-être public,Aidez les apprenants PHP à grandir rapidement!