Машинное обучение с TensorFlow: создание первых моделей на Python
2026-02-26 20:08 Diff
https://ru.hexlet.io/blog/posts/mashinnoe-obuchenie-s-tensorflow-sozdanie-pervyh-modeley-na-python

Рассказываем о библиотеке TensorFlow и показываем, как создать первые модели машинного обучения с использованием TensorFlow на Python.

Содержание

Что такое TensorFlow

Библиотека TensorFlow предназначена для работы с числами и для создания моделей машинного обучения. Задача библиотеки — помогать в обучении искусственного интеллекта. Изначально TensorFlow разрабатывалась для Python, хотя есть версии и для других языков.

Кроме открытого исходного кода, у TensorFlow есть и другие преимущества:

  • Поддерживает распределенные вычисления, что позволяет обрабатывать большие объемы данных быстрее, используя несколько компьютеров или процессоров одновременно.
  • TensorFlow может работать на разных устройствах, таких как обычные процессоры (CPU), графические процессоры (GPU) и специальные процессоры для машинного обучения, например Tensor Processor Unit.
  • У TensorFlow есть большое и активное сообщество разработчиков, которые создают учебные материалы, отвечают на вопросы и помогают друг другу.

Установка и настройка TensorFlow

Перед установкой TensorFlow убедитесь, что у вас установлены Python версии 3.6–3.9 и pip.

Для установки используйте команду:

После этого запустите Python и импортируйте TensorFlow:

Основные концепции TensorFlow

Приступая к работе с TensorFlow, будьте уверены, что знакомы с основными терминами и понятиями, необходимыми для работы с этой библиотекой.

Тензоры: многомерные массивы данных

Тензоры представляют собой многомерные массивы в TensorFlow. Одномерный тензор может представлять собой вектор, двумерный — матрицу, а тензор с большим числом измерений может использоваться для представления сложных структур данных, таких как изображения или временные ряды.

Операции с тензорами

В TensorFlow возможны разные типы операций с тензорами, например стандартные математические функции, такие как сложение и вычитание, и более сложные операции: матричное умножение, вычисление свертки и логические операции.

Граф вычислений

Граф вычислений в TensorFlow — это набор операций и тензоров, связанных между собой. Граф определяет, каким образом будут проводиться вычисления, каков порядок выполнения операций и зависимость между ними.

Сессии

Сессии в TensorFlow обеспечивают выполнение операций, определенных в графе вычислений. Сначала создается граф, который определяет вычисления, а затем запускается сессия, где выполняется граф или часть графа. Каждую сессию необходимо закрыть.

Keras — высокоуровневый API

Keras — это встроенный в TensorFlow высокоуровневый API с удобным интерфейсом для определения слоев нейронных сетей, настройки параметров обучения и управления процессом обучения. С помощью Keras можно быстро создавать сложные модели, используя всего несколько строк кода.

Создание простой модели на Python

Чтобы создать простую модель на Python, выполните следующие шаги.

Подготовка данных

Для начала нужно подготовить данные, которые будут использоваться для обучения модели, например 1000 строк данных по 10 признаков (labels) на каждую, с помощью библиотеки NumPy.

Построение модели

Затем с помощью API Keras для TensorFlow строится сама модель, состоящая из нужного числа слоев, например последовательная модель (Sequential), состоящая из двух слоев: один скрытый и один выходной.

Компиляция модели

Компиляция включает выбор оптимизатора, функции потерь и метрик, которые будут использоваться для оценки качества модели.

Обучение модели

На заключительном этапе мы обучаем модель на подготовленных данных. Эпохи — это число полных проходов по набору данных, а батч — число образцов, используемых для одной итерации весовых коэффициентов модели.

Оценка и улучшение модели

После создания модели ее можно улучшать, оценив точность на тестовой выборке данных. Для этого нужно последовательно выполнить такие шаги:

  • Разделение данных на обучающую и тестовую выборки.
  • Обучение модели на обучающей части выборки.
  • Оценка точности модели на тестовой выборке.
  • Выбор метода улучшения модели.

Можно воспользоваться следующими методами улучшения модели:

  • Изменение архитектуры, то есть другие значения числа строк и признаков (labels).
  • Регуляризация и Dropout, то есть предотвращение переобучения.
  • Изменения гиперпараметров — количества эпох и размера батча.

Заключение

Создать простую модель для машинного обучения, настроить ее, обучить и улучшить несложно, если воспользоваться библиотекой TensorFlow для языка Python.