Loss

class KGE.loss.Loss[source]

Bases: object

A base module for loss.

Methods Summary

__call__(pos_score, neg_score)

Calculate loss.

Methods Documentation

__call__(pos_score, neg_score)[source]

Calculate loss.

Parameters

pos_score (tf.Tensor) – score of postive triplets, with shape (n,)
neg_score (tf.Tensor) – score of negative triplets, with shape (n,)

__init__()[source]: Initialize loss

__new__(*args, **kwargs)