tf.contrib.learn.Estimator.__init__(model_fn=None, model_dir=None, config=None, params=None, feature_engineering_fn=None)
Constructs an Estimator instance.
Args:
model_fn: Model function, takes features and targets tensors or dicts of tensors and returns predictions and loss tensors. Supports next three signatures for the function: (features, targets) -> (predictions, loss, train_op) (features, targets, mode) -> (predictions, loss, train_op) (features, targets, mode, params) -> (pred