After checking, the parameters for classification are:
message AdamOptimizerConfig {
float lr = 1;
float beta_1 = 2;
float beta_2 = 3;
float epsilon = 4;
float decay = 5;
}
So,
optimizer {
adam {
epsilon: 1e-7
beta_1: 0.9
beta_2: 0.999
lr: 0.01 (just an example)
decay: 0.0
}
}