MCPcopy
hub / github.com/shenweichen/DeepCTR-Torch / compile

Method compile

deepctr_torch/models/basemodel.py:439–451  ·  view source on GitHub ↗

:param optimizer: String (name of optimizer) or optimizer instance. See [optimizers](https://pytorch.org/docs/stable/optim.html). :param loss: String (name of objective function) or objective function. See [losses](https://pytorch.org/docs/stable/nn.functional.html#loss-functions).

(self, optimizer,
                loss=None,
                metrics=None,
                )

Source from the content-addressed store, hash-verified

437 self.aux_loss = aux_loss * alpha
438
439 def compile(self, optimizer,
440 loss=None,
441 metrics=None,
442 ):
443 """
444 :param optimizer: String (name of optimizer) or optimizer instance. See [optimizers](https://pytorch.org/docs/stable/optim.html).
445 :param loss: String (name of objective function) or objective function. See [losses](https://pytorch.org/docs/stable/nn.functional.html#loss-functions).
446 :param metrics: List of metrics to be evaluated by the model during training and testing. Typically you will use `metrics=['accuracy']`.
447 """
448 self.metrics_names = ["loss"]
449 self.optim = self._get_optim(optimizer)
450 self.loss_func = self._get_loss_func(loss)
451 self.metrics = self._get_metrics(metrics)
452
453 def _get_optim(self, optimizer):
454 if isinstance(optimizer, str):

Callers 13

check_modelFunction · 0.80
check_mtl_modelFunction · 0.80
test_MLRsFunction · 0.80
test_MLRFunction · 0.80
run_din.pyFile · 0.80
run_dien.pyFile · 0.80

Calls 3

_get_optimMethod · 0.95
_get_loss_funcMethod · 0.95
_get_metricsMethod · 0.95

Tested by 5

test_MLRsFunction · 0.64
test_MLRFunction · 0.64