CalibrationLayer

class ampligraph.latent_features.layers.calibration.CalibrationLayer(*args: Any, **kwargs: Any)

Layer to calibrate the model outputs.

The class implements the heuristics described in [TC20], using Platt scaling [].

See the docs of calibrate() for more details.

Attributes

Methods

__init__([pos_size, neg_size, ...])
build(input_shape)	Build method.
call(scores_pos[, scores_neg, training])	Call method.
get_config()

__init__(pos_size=0, neg_size=0, positive_base_rate=None, **kwargs)

build(input_shape)

Build method.

call(scores_pos, scores_neg=[], training=0)

Call method.

get_config()