Spec-Zone.ru › TensorFlow 2.4

tf.compat.v1.train.ProximalGradientDescentOptimizer

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

Наследуется от: Optimizer

tf.compat.v1.train.ProximalGradientDescentOptimizer(
    learning_rate, l1_regularization_strength=0.0, l2_regularization_strength=0.0,
    use_locking=False, name='ProximalGradientDescent'
)

Ссылки:

Эффективное обучение с использованием разбиения вперёд-назад: Duchi et al., 2009 (pdf)

Аргументы
learning_rate Тензор или значение с плавающей точкой. Скорость обучения.
l1_regularization_strength Вещественное значение, должно быть больше или равно нулю.
l2_regularization_strength Вещественное значение, должно быть больше или равно нулю.
use_locking Если True, использовать блокировки для операций обновления.
name Необязательный префикс имени для операций, создаваемых при применении градиентов. По умолчанию "GradientDescent".

Методы

apply_gradients

Просмотреть исходный код

apply_gradients(
    grads_and_vars, global_step=None, name=None
)

Применить градиенты к переменным.

Это вторая часть minimize(). Она возвращает Operation, который применяет градиенты.

Аргументы
grads_and_vars Список пар (градиент, переменная), возвращаемых методом compute_gradients().
global_step Необязательное Variable, увеличиваемое на единицу после обновления переменных.
name Необязательное имя возвращаемой операции. По умолчанию используется имя, переданное конструктору Optimizer.
Возвращает
Operation, применяющий заданные градиенты. Если global_step не равно None, эта операция также увеличивает global_step.
Исключения
TypeError Если grads_and_vars имеет неправильный формат.
ValueError Если ни у одной из переменных нет градиентов.
RuntimeError Если нужно использовать _distributed_apply() вместо этого.

compute_gradients

Просмотреть исходный код

compute_gradients(
    loss, var_list=None, gate_gradients=GATE_OP, aggregation_method=None,
    colocate_gradients_with_ops=False, grad_loss=None
)

Вычислить градиенты loss для переменных в var_list.

Это первая часть minimize(). Она возвращает список пар (градиент, переменная), где «градиент» — градиент для «переменной». Обратите внимание, что «градиент» может быть Tensor, IndexedSlices или None, если для данной переменной нет градиента.

Аргументы
loss Тензор, содержащий значение для минимизации, или вызываемый без аргументов, возвращающий значение для минимизации. В режиме выполнения eager он должен быть вызываемым.
var_list Необязательный список или кортеж tf.Variable для обновления, чтобы минимизировать loss. По умолчанию используется список переменных, собранных в графе по ключу GraphKeys.TRAINABLE_VARIABLES.
gate_gradients Способ управления вычислением градиентов. Может быть GATE_NONE, GATE_OP или GATE_GRAPH.
aggregation_method Указывает метод комбинирования градиентов. Допустимые значения определены в классе AggregationMethod.
colocate_gradients_with_ops Если True, попробовать разместить градиенты вместе с соответствующей операцией.
grad_loss Необязательно. Tensor содержащий градиент, вычисленный для loss.
Возвращает
Список пар (градиент, переменная). Переменная всегда присутствует, но градиент может быть None.
Исключения
TypeError Если var_list содержит что-либо кроме объектов Variable.
ValueError Если некоторые аргументы некорректны.
RuntimeError Если вызвано с включенным eager выполнением и loss не является вызываемым.

Совместимость с eager выполнением

При включённом режиме eager выполнения gate_gradients, aggregation_method, и colocate_gradients_with_ops игнорируются.

get_name

Просмотреть исходный код

get_name()

get_slot

Просмотреть исходный код

get_slot(
    var, name
)

Возвращает слот с именем name, созданный для var оптимизатором.

Некоторые подклассы Optimizer используют дополнительные переменные. Например, Momentum и Adagrad используют переменные для накопления обновлений. Этот метод даёт доступ к этим Variable объектам, если они нужны по какой-то причине.

Используйте get_slot_names() для получения списка имён слотов, созданных Optimizer.

Аргументы
var Переменная, переданная в minimize() или apply_gradients().
name Строка.
Возвращает
Variable для слота, если он был создан, None в противном случае.

get_slot_names

Просмотреть исходный код

get_slot_names()

Возвращает список имён слотов, созданных Optimizer.

См. get_slot().

Возвращает
Список строк.

minimize

Просмотреть исходный код

minimize(
    loss, global_step=None, var_list=None, gate_gradients=GATE_OP,
    aggregation_method=None, colocate_gradients_with_ops=False, name=None,
    grad_loss=None
)

Добавить операции для минимизации loss путём обновления var_list.

Этот метод просто объединяет вызовы compute_gradients() и apply_gradients(). Если нужно обработать градиент перед применением, вызовите compute_gradients() и apply_gradients() явно вместо использования этой функции.

Аргументы
loss Tensor содержащий значение для минимизации.
global_step Необязательное Variable увеличиваемое на единицу после обновления переменных.
var_list Необязательный список или кортеж объектов Variable для обновления, чтобы минимизировать loss. По умолчанию используется список переменных, собранных в графе по ключу GraphKeys.TRAINABLE_VARIABLES.
gate_gradients Способ управления вычислением градиентов. Может быть GATE_NONE, GATE_OP или GATE_GRAPH.
aggregation_method Указывает метод комбинирования градиентов. Допустимые значения определены в классе AggregationMethod.
colocate_gradients_with_ops Если True, попробовать разместить градиенты вместе с соответствующей операцией.
name Необязательное имя возвращаемой операции.
grad_loss Необязательно. Tensor содержащий градиент, вычисленный для loss.
Возвращает
Операция, обновляющая переменные в var_list. Если global_step не было None, эта операция также увеличивает global_step.
Исключения
ValueError Если некоторые переменные не являются объектами Variable.

Совместимость с eager выполнением

При включенном режиме eager выполнения loss должен быть Python-функцией без аргументов, вычисляющей значение для минимизации. Минимизация (и вычисление градиента) выполняется относительно элементов var_list, если оно не равно None, иначе — относительно всех обучаемых переменных, созданных во время выполнения функции loss. gate_gradients, aggregation_method, colocate_gradients_with_ops и grad_loss игнорируются при включённом eager режиме.

variables

Просмотреть исходный код

variables()

Список переменных, которые кодируют текущее состояние Optimizer.

Включает переменные слотов и дополнительные глобальные переменные, созданные оптимизатором в текущей стандартной графе.

class="responsive fixed orange">
Возвращаемое значение
Список переменных.
class="responsive fixed orange">
Переменные класса
GATE_GRAPH 2
GATE_NONE 0
GATE_OP 1

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/train/ProximalGradientDescentOptimizer

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API