(self)
| 329 | |
| 330 | # compute loss |
| 331 | def _create_loss(self): |
| 332 | loss_func = tf.keras.losses.CategoricalCrossentropy() |
| 333 | self.probability = tf.squeeze(self.probability) |
| 334 | self.loss = tf.math.reduce_mean( |
| 335 | loss_func(self._label, self.probability)) |
| 336 | tf.summary.scalar('loss', self.loss) |
| 337 | |
| 338 | # define optimizer and generate train_op |
| 339 | def _create_optimizer(self): |
no test coverage detected