(self, inputs)
| 41 | return self.encoder(n_id) |
| 42 | |
| 43 | def __call__(self, inputs): |
| 44 | label = self.get_label_fn(inputs) |
| 45 | embedding = self.embed(inputs) |
| 46 | logit = self.logit_fn(embedding) |
| 47 | |
| 48 | metric = self.metric_class(label, tf.nn.sigmoid(logit)) |
| 49 | loss = self.loss_fn(label, logit) |
| 50 | return (embedding, loss, self.metric_name, metric) |