LpRegularizer

class LpRegularizer(device, weight=1.0, dim=- 1, normalize=False, p=2.0, apply_only_once=False)[source]

Bases: pykeen.regularizers.Regularizer

A simple L_p norm based regularizer.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

Attributes Summary

hpo_default

The default strategy for optimizing the regularizer’s hyper-parameters

Methods Summary

forward(x)

Compute the regularization term for one tensor.

Attributes Documentation

hpo_default: ClassVar[Mapping[str, Any]] = {'weight': {'high': 1.0, 'low': 0.01, 'scale': 'log', 'type': <class 'float'>}}

The default strategy for optimizing the regularizer’s hyper-parameters

Methods Documentation

forward(x)[source]

Compute the regularization term for one tensor.

Return type

FloatTensor