Calls the model_fn with required parameters.
(self, features, labels, mode, params)
| 177 | return eval_step, captured_eval_metric_fn, captured_scaffold_fn |
| 178 | |
| 179 | def _call_model_fn(self, features, labels, mode, params): |
| 180 | """Calls the model_fn with required parameters.""" |
| 181 | model_fn_args = function_utils.fn_args(self._model_fn) |
| 182 | kwargs = {} |
| 183 | |
| 184 | if 'labels' in model_fn_args: |
| 185 | kwargs['labels'] = labels |
| 186 | elif labels is not None: |
| 187 | raise ValueError( |
| 188 | 'model_fn does not take labels, but input_fn returns labels.') |
| 189 | if 'mode' in model_fn_args: |
| 190 | kwargs['mode'] = mode |
| 191 | |
| 192 | if 'params' in model_fn_args: |
| 193 | kwargs['params'] = params |
| 194 | |
| 195 | return self._verify_estimator_spec( |
| 196 | self._model_fn(features=features, **kwargs)) |
| 197 | |
| 198 | def _verify_estimator_spec(self, estimator_spec): |
| 199 | """Verifies estimator spec contains correct data.""" |
no test coverage detected