So I am using
tf.estimator.Estimator(
model_fn, model_dir=None, config=None, params=None, warm_start_from=None
)
and I am confused about the parameter params
.
I know it's a dict
and according to some example code, I assume that params
is some like:
params = {"batch_size":128,
"hidden_layer": 3
}
but according to the official page, params
is dict of hyper parameters that will be passed into model_fn. Keys are names of parameters, values are basic python types (offical page). So the value should be python types like int64, float64?
Please give me a clear explanation. Thank you so much for your help
from Confusion about the parameter in tf.estimator.Estimator
No comments:
Post a Comment