Home | History | Annotate | Download | only in estimators

Lines Matching refs:features

59 def _get_feature_dict(features):
60 if isinstance(features, dict):
61 return features
62 return {"": features}
65 def debug_model_fn(features, labels, mode, params, config=None):
69 features: `Tensor` or dict of `Tensor` (depends on data passed to `fit`).
79 ValueError: If features is an empty dictionary.
86 features = _get_feature_dict(features)
87 if not features:
88 raise ValueError("Features cannot be empty.")
95 # among all of the features.
96 for feature in features.values():
111 features=features,
148 Input of `fit` and `evaluate` should have following features,
176 feature_engineering_fn: Feature engineering function. Takes features and
178 features and labels which will be fed into the model.
202 """Returns predicted classes for given features.
220 """Returns prediction probabilities for given features.
264 Input of `fit` and `evaluate` should have following features,
290 feature_engineering_fn: Feature engineering function. Takes features and
292 features and labels which will be fed into the model.
313 """Returns predicted scores for given features.