@roxanne.hauck
Для создания и выполнения простой программы на TensorFlow, вам нужно установить библиотеку, импортировать необходимые модули и создать модель для обучения. Давайте рассмотрим пример программы для классификации изображений с использованием TensorFlow:
- Установите TensorFlow на ваш компьютер. Вы можете установить TensorFlow, выполнив следующую команду:
pip install tensorflow
- Импортируйте необходимые библиотеки TensorFlow и NumPy:
import tensorflow as tf
from tensorflow.keras import datasets, layers, models
import numpy as np
- Загрузите набор данных, например, набор данных MNIST (набор изображений рукописных цифр) с помощью TensorFlow:
(train_images, train_labels), (test_images, test_labels) = datasets.mnist.load_data()
- Нормализуйте данные и преобразуйте метки в формат one-hot encoding:
train_images = train_images / 255.0
test_images = test_images / 255.0
train_labels = tf.keras.utils.to_categorical(train_labels, num_classes=10)
test_labels = tf.keras.utils.to_categorical(test_labels, num_classes=10)
- Создайте модель нейронной сети с помощью библиотеки Keras:
model = models.Sequential()
model.add(layers.Flatten(input_shape=(28, 28)))
model.add(layers.Dense(128, activation='relu'))
model.add(layers.Dense(10, activation='softmax'))
- Скомпилируйте модель, указав оптимизатор, функцию потерь и метрику:
model.compile(optimizer='adam',
loss='categorical_crossentropy',
metrics=['accuracy'])
- Обучите модель на обучающем наборе данных:
model.fit(train_images, train_labels, epochs=5)
- Оцените производительность модели на тестовом наборе данных:
test_loss, test_accuracy = model.evaluate(test_images, test_labels)
print(f'Test accuracy: {test_accuracy}')
Это простой пример программы на TensorFlow для классификации изображений с использованием нейронной сети. Вы можете доработать этот пример, добавив больше слоев, изменяя гиперпараметры и т.д. для решения более сложных задач.