Модуль: tf.compat.v1.train

Поддержка обучения моделей.

См. руководство по обучению.

Модули

experimental модуль: Публичный API для пространства имён tf.train.experimental.

queue_runner модуль: Публичный API для пространства имён tf.train.queue_runner.

Классы

class AdadeltaOptimizer: Оптимизатор, реализующий алгоритм Adadelta.

class AdagradDAOptimizer: Алгоритм Adagrad Dual Averaging для разреженных линейных моделей.

class AdagradOptimizer: Оптимизатор, реализующий алгоритм Adagrad.

class AdamOptimizer: Оптимизатор, реализующий алгоритм Adam.

class BytesList: Протокол сообщений

class Checkpoint: Объединяет отслеживаемые объекты, сохраняя и восстанавливая их.

class CheckpointManager: Управляет несколькими контрольными точками, сохраняя некоторые и удаляя ненужные.

class CheckpointOptions: Параметры для создания контрольной точки.

class CheckpointSaverHook: Сохраняет контрольные точки каждые N шагов или секунд.

class CheckpointSaverListener: Интерфейс для слушателей, которые выполняют действия до или после сохранения контрольной точки.

class ChiefSessionCreator: Создаёт tf.compat.v1.Session для руководителя.

class ClusterDef: Протокол сообщений

class ClusterSpec: Представляет кластер как набор «задач», организованных в «работы».

class Coordinator: Координатор для потоков.

class Example: Протокол сообщений

class ExponentialMovingAverage: Поддерживает скользящие средние значения переменных с использованием экспоненциального затухания.

class Feature: Протокол сообщений

class FeatureList: Протокол сообщений

class FeatureLists: Протокол сообщений

class Features: Протокол сообщений

class FeedFnHook: Выполняет feed_fn и устанавливает feed_dict соответственно.

class FinalOpsHook: Плагин, который оценивает Tensors в конце сеанса.

class FloatList: Протокол сообщений

class FtrlOptimizer: Оптимизатор, реализующий алгоритм FTRL.

class GlobalStepWaiterHook: Откладывает выполнение до тех пор, пока глобальный шаг не достигнет wait_until_step.

class GradientDescentOptimizer: Оптимизатор, реализующий алгоритм градиентного спуска.

class Int64List: Протокол сообщений

class JobDef: Протокол сообщений

class LoggingTensorHook: Выводит указанные тензоры каждые N локальных шагов, каждые N секунд или в конце.

class LooperThread: Поток, который многократно выполняет код, необязательно по таймеру.

class MomentumOptimizer: Оптимизатор, реализующий алгоритм импульса.

class MonitoredSession: Объект, похожий на сеанс, который обрабатывает инициализацию, восстановление и плагины.

class NanLossDuringTrainingError: Неопределённая ошибка времени выполнения.

class NanTensorHook: Отслеживает тензор потерь и останавливает обучение, если потери равны NaN.

class Optimizer: Базовый класс для оптимизаторов.

class ProfilerHook: Захватывает информацию о профилировании CPU/GPU каждые N шагов или секунд.

class ProximalAdagradOptimizer: Оптимизатор, реализующий алгоритм Proximal Adagrad.

class ProximalGradientDescentOptimizer: Оптимизатор, реализующий алгоритм приближённого градиентного спуска.

class QueueRunner: Содержит список операций очереди для добавления, каждая из которых выполняется в потоке.

class RMSPropOptimizer: Оптимизатор, реализующий алгоритм RMSProp (Tielemans и др.)

class Saver: Сохраняет и восстанавливает переменные.

class SaverDef: Протокол сообщений

class Scaffold: Структура для создания или сбора компонентов, обычно необходимых для обучения модели.

class SecondOrStepTimer: Таймер, срабатывающий не более одного раза каждые N секунд или каждые N шагов.

class SequenceExample: Протокол сообщений

class Server: Встроенный сервер TensorFlow для использования в распределённом обучении.

class ServerDef: Протокол сообщений

class SessionCreator: Фабрика для tf.Session.

class SessionManager: Помощник обучения, восстанавливающий данные из контрольной точки и создающий сеанс.

class SessionRunArgs: Представляет аргументы, которые необходимо добавить к вызову Session.run().

class SessionRunContext: Предоставляет информацию о вызове session.run().

class SessionRunHook: Плагин для расширения вызовов MonitoredSession.run().

class SessionRunValues: Содержит результаты вызова Session.run().

class SingularMonitoredSession: Объект, похожий на сеанс, который обрабатывает инициализацию, восстановление и плагины.

class StepCounterHook: Плагин, который подсчитывает шаги в секунду.

class StopAtStepHook: Плагин, который запрашивает остановку на указанном шаге.

class SummarySaverHook: Сохраняет сводки каждые N шагов.

class Supervisor: Помощник обучения, который сохраняет модели и вычисляет сводки.

class SyncReplicasOptimizer: Класс для синхронизации, агрегации градиентов и передачи их оптимизатору.

class VocabInfo: Информация о словаре для теплого запуска.

class WorkerSessionCreator: Создаёт tf.compat.v1.Session для рабочего.

Функции

MonitoredTrainingSession(...): Создаёт MonitoredSession для обучения.

NewCheckpointReader(...): Функция, возвращающая CheckPointReader.

add_queue_runner(...): Добавляет QueueRunner в коллекцию в графе. (устарело)

assert_global_step(...): Утверждает, что global_step_tensor является скалярным целым числом Variable или Tensor.

basic_train_loop(...): Основной цикл для обучения модели.

batch(...): Создаёт пакеты тензоров в tensors. (устарело)

batch_join(...): Выполняет список тензоров для заполнения очереди, чтобы создать пакеты примеров. (устарело)

checkpoint_exists(...): Проверяет, существует ли контрольная точка V1 или V2 с указанным префиксом. (устарело)

checkpoints_iterator(...): Непрерывно предоставляет новые файлы контрольных точек по мере их появления.

cosine_decay(...): Применяет к скорости обучения расчёт по косинусной функции.

cosine_decay_restarts(...): Применяет к скорости обучения расчёт по косинусной функции с перезапусками.

create_global_step(...): Создаёт тензор глобального шага в графе.

do_quantize_training_on_graphdef(...): Разрабатывается общая схема квантования в tf.contrib.quantize. (устарело)

exponential_decay(...): Применяет экспоненциальное затухание к скорости обучения.

export_meta_graph(...): Возвращает MetaGraphDef протокол.

generate_checkpoint_state_proto(...): Генерирует протокол состояния контрольной точки.

get_checkpoint_mtimes(...): Возвращает mtimes (отметки времени изменения) контрольных точек. (устарело)

get_checkpoint_state(...): Возвращает CheckpointState протокол из файла "checkpoint".

get_global_step(...): Получение тензора глобального шага.

get_or_create_global_step(...): Возвращает и создаёт (при необходимости) тензор глобального шага.

global_step(...): Небольшая вспомогательная функция для получения глобального шага.

import_meta_graph(...): Восстанавливает граф, сохранённый в MetaGraphDef протоколе.

init_from_checkpoint(...): Заменяет инициализаторы tf.Variable, чтобы они загружали данные из файла контрольной точки.

input_producer(...): Выводит строки input_tensor в очередь для входной обработки. (устарело)

inverse_time_decay(...): Применяет обратное затухание по времени к начальной скорости обучения.

latest_checkpoint(...): Находит имя файла последней сохранённой контрольной точки.

limit_epochs(...): Возвращает тензор num_epochs раз, а затем вызывает ошибку OutOfRange. (устарело)

linear_cosine_decay(...): Применяет линейное косинусное затухание к скорости обучения.

list_variables(...): Список ключей контрольной точки и форм переменных в контрольной точке.

load_checkpoint(...): Возвращает CheckpointReader для контрольной точки, найденной в ckpt_dir_or_file.

load_variable(...): Возвращает значение тензора заданной переменной в контрольной точке.

match_filenames_once(...): Сохраняет список файлов, соответствующих шаблону, чтобы вычисление производилось только один раз.

maybe_batch(...): Условно создает пакеты тензоров на основе keep_input. (устарело)

maybe_batch_join(...): Выполняет список тензоров для условного заполнения очереди, чтобы создать пакеты. (устарело)

maybe_shuffle_batch(...): Создает пакеты, случайным образом перемешивая условно вставленные в очередь тензоры. (устарело)

maybe_shuffle_batch_join(...): Создает пакеты, случайным образом перемешивая условно вставленные в очередь тензоры. (устарело)

natural_exp_decay(...): Применяет естественную экспоненциальную деградацию к начальной скорости обучения.

noisy_linear_cosine_decay(...): Применяет шумную линейную деградацию косинусоиды к скорости обучения.

piecewise_constant(...): Кусочно-постоянная функция от границ и значений интервалов.

piecewise_constant_decay(...): Кусочно-постоянная функция от границ и значений интервалов.

polynomial_decay(...): Применяет полиномиальную деградацию к скорости обучения.

range_input_producer(...): Выдаёт целые числа от 0 до limit-1 в очереди. (устарело)

remove_checkpoint(...): Удаляет контрольную точку, заданную checkpoint_prefix. (устарело)

replica_device_setter(...): Возвращает device function для использования при построении графа для реплик.

sdca_fprint(...): Вычисляет отпечатки пальцев входных строк.

sdca_optimizer(...): Распределённая версия оптимизатора стохастического двойственного восхождения по координатам (SDCA) для

sdca_shrink_l1(...): Применяет L1 регуляризацию shrink к параметрам.

shuffle_batch(...): Создает пакеты, случайным образом перемешивая тензоры. (устарело)

shuffle_batch_join(...): Создает пакеты, случайным образом перемешивая тензоры. (устарело)

slice_input_producer(...): Выдаёт срез каждого Tensor в tensor_list. (устарело)

start_queue_runners(...): Запускает все исполнители очереди, собранные в графе. (устарело)

string_input_producer(...): Выводит строки (например, имена файлов) в очередь для входной обработки данных. (устарело)

summary_iterator(...): Возвращает итератор для чтения Event протоколов из файла событий.

update_checkpoint_state(...): Обновляет содержимое файла 'checkpoint'. (устарело)

warm_start(...): Теплое начало модели с использованием заданных настроек.

write_graph(...): Записывает протокол графа в файл.