自定义SGD优化器。

def create_hash_optimizer(learning_rate, use_locking=False, name="GradientDescent")

参数名	类型	必选/可选	说明
learning_rate	float/tf.Tensor	必选	学习率。默认值：无取值范围：[0.0, 10.0]
use_locking	bool	可选	优化器中防止对变量并发更新。默认值：False 取值范围： True False
name	string	可选	优化器名称。默认值：GradientDescent 名称长度范围：[1, 200]

参数名

类型

必选/可选

说明

learning_rate

float/tf.Tensor

必选

学习率。

默认值：无

取值范围：[0.0, 10.0]

use_locking

bool

可选

优化器中防止对变量并发更新。

默认值：False

取值范围：

name

string

可选

优化器名称。

默认值：GradientDescent

名称长度范围：[1, 200]

CustomizedGradientDescent（自定义SGD优化器）的一个实例对象。

from mx_rec.optimizers.gradient_descent import create_hash_optimizer
hashtable_optimizer = create_hash_optimizer(0.001)