Method
__init__
(self,
get_label_fn,
encoder_fn,
logit_fn,
metric_name='f1',
loss_fn=sigmoid_loss)
Source from the content-addressed store, hash-verified
| 25 | |
| 26 | class SuperviseSolution(object): |
| 27 | def __init__(self, |
| 28 | get_label_fn, |
| 29 | encoder_fn, |
| 30 | logit_fn, |
| 31 | metric_name='f1', |
| 32 | loss_fn=sigmoid_loss): |
| 33 | self.get_label_fn = get_label_fn |
| 34 | self.metric_name = metric_name |
| 35 | self.metric_class = tf_euler.utils.metrics.get(metric_name) |
| 36 | self.encoder = encoder_fn |
| 37 | self.logit_fn = logit_fn |
| 38 | self.loss_fn = loss_fn |
| 39 | |
| 40 | def embed(self, n_id): |
| 41 | return self.encoder(n_id) |
Callers
nothing calls this directly
Tested by
no test coverage detected