(self,
parse_input_fn,
encoder_fn,
parse_group_emb_fn,
logit_fn=CosineLogits(),
metric_name='auc',
neg_sample_fn=None,
loss_fn=sigmoid_loss)
| 28 | |
| 29 | class SuperviseSampleSolution(object): |
| 30 | def __init__(self, |
| 31 | parse_input_fn, |
| 32 | encoder_fn, |
| 33 | parse_group_emb_fn, |
| 34 | logit_fn=CosineLogits(), |
| 35 | metric_name='auc', |
| 36 | neg_sample_fn=None, |
| 37 | loss_fn=sigmoid_loss): |
| 38 | self.metric_name = metric_name |
| 39 | self.metric_class = tf_euler.utils.metrics.get(metric_name) |
| 40 | self.parse_input_fn = parse_input_fn |
| 41 | self.parse_group_emb_fn = parse_group_emb_fn |
| 42 | self.encoder = encoder_fn |
| 43 | self.neg_sample_fn = neg_sample_fn |
| 44 | self.logit_fn = logit_fn |
| 45 | self.loss_fn = loss_fn |
| 46 | |
| 47 | def embed(self, n_id): |
| 48 | return self.encoder(n_id) |
nothing calls this directly
no test coverage detected