SGDByAddr
自定义SGDByAddr地址优化器。
函数原型
def create_hash_optimizer_by_addr(learning_rate, weight_decay=0.0001, use_locking=False, name="GradientDescentByAddr")
参数说明
参数名称 |
参数说明 |
参数类型 |
必选/可选 |
默认值 |
取值范围 |
|---|---|---|---|---|---|
learning_rate |
学习率。 |
float |
必选 |
无 |
[-2147483647.0, 2147483647.0] |
weight_decay |
权重衰退。 |
float |
可选 |
0.0001 |
[0.0, 1.0] |
use_locking |
优化器中防止对变量并发更新。 |
bool |
可选 |
False |
|
name |
优化器名称。 |
string |
可选 |
GradientDescentByAddr |
优化器名称长度范围：[1, 255] |
返回值说明
CustomizedGradientDescentByAddr（自定义SGD地址优化器）的一个实例对象。
使用示例
from mx_rec.optimizers.gradient_descent_by_addr import create_hash_optimizer_by_addr hashtable_optimizer = create_hash_optimizer_by_addr(0.001)
父主题： 优化器