BCELoss

class ampligraph.latent_features.BCELoss(eta, loss_params={}, verbose=False)

Binary Cross Entropy Loss.

\[\mathcal{L} = - \frac{1}{N} \sum_{i=1}^{N} y_i \cdot log(p(y_i)) + (1-y_i) \cdot log(1-p(y_i))\]

Examples

>>> from ampligraph.latent_features.models import ConvE
>>> model = ConvE(batches_count=1, seed=555, epochs=20, k=10, loss='bce', loss_params={})

Methods

__init__(eta[, loss_params, verbose])

Initialize Loss

__init__(eta, loss_params={}, verbose=False)

Initialize Loss

Parameters

loss_params (dict) – Dictionary of loss-specific hyperparams: