r'''Constructs a wrapped `Estimator` instance. Args: model_fn: Model function. See `tensorflow_estimator/python/estimator/estimator.py#L145` for more information. kwargs: Estimator arguments.
(self, model_fn, **kwargs)
| 1060 | ''' |
| 1061 | |
| 1062 | def __init__(self, model_fn, **kwargs): |
| 1063 | r'''Constructs a wrapped `Estimator` instance. |
| 1064 | |
| 1065 | Args: |
| 1066 | model_fn: Model function. See |
| 1067 | `tensorflow_estimator/python/estimator/estimator.py#L145` |
| 1068 | for more information. |
| 1069 | kwargs: Estimator arguments. |
| 1070 | ''' |
| 1071 | model_dir = kwargs.get('model_dir', None) |
| 1072 | self._train_drop_remainder = kwargs.pop( |
| 1073 | 'train_drop_remainder', True) |
| 1074 | self._eval_drop_remainder = kwargs.pop('eval_drop_remainder', True) |
| 1075 | self._predict_drop_remainder = kwargs.pop( |
| 1076 | 'predict_drop_remainder', True) |
| 1077 | |
| 1078 | super().__init__( |
| 1079 | wraps_model_fn(model_fn, model_dir, kwargs['config']), |
| 1080 | **kwargs) |
| 1081 | |
| 1082 | def _assert_members_are_not_overridden(self): |
| 1083 | r'''disable the overridden check here. |
nothing calls this directly
no test coverage detected