Args: optimizer (tf.train.Optimizer): loss, metrics: string or list of strings
(self, optimizer, loss, metrics=None)
| 254 | self.trainer = trainer |
| 255 | |
| 256 | def compile(self, optimizer, loss, metrics=None): |
| 257 | """ |
| 258 | Args: |
| 259 | optimizer (tf.train.Optimizer): |
| 260 | loss, metrics: string or list of strings |
| 261 | """ |
| 262 | if isinstance(loss, six.string_types): |
| 263 | loss = [loss] |
| 264 | if metrics is None: |
| 265 | metrics = [] |
| 266 | if isinstance(metrics, six.string_types): |
| 267 | metrics = [metrics] |
| 268 | |
| 269 | self._stats_to_inference = loss + metrics + [TOTAL_LOSS_NAME] |
| 270 | setup_keras_trainer( |
| 271 | self.trainer, get_model=self.get_model, |
| 272 | input_signature=self.input_signature, |
| 273 | target_signature=self.target_signature, |
| 274 | input=self.input, |
| 275 | optimizer=optimizer, |
| 276 | loss=loss, |
| 277 | metrics=metrics) |
| 278 | |
| 279 | def fit(self, validation_data=None, **kwargs): |
| 280 | """ |
no test coverage detected