Saturday, 22 August 2020

Confusion about the parameter in tf.estimator.Estimator

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