Method __call__

tf_euler/python/solution/base_sample.py:127–136

(self, inputs)

        return src, pos, neg

    def __call__(self, inputs):
        src, pos, negs = self.to_sample(inputs)
        embedding = self.target_embed(src)
        embedding_pos = self.context_embed(pos)
        embedding_negs = self.context_embed(negs)

        logits, neg_logits = self.logit_fn(embedding, embedding_pos, embedding_negs)
        loss = self.loss_fn(logits, neg_logits)
        metric = self.metric_fn(logits, neg_logits)
        return (embedding, loss, self.metric_name, metric)