SLCWATrainingLoop
- class SLCWATrainingLoop(negative_sampler=None, negative_sampler_kwargs=None, **kwargs)[source]
Bases:
pykeen.training.training_loop.TrainingLoop
[Tuple
[torch.LongTensor
,torch.LongTensor
,Optional
[torch.BoolTensor
]],pykeen.triples.instances.SLCWABatch
]A training loop that uses the stochastic local closed world assumption training approach.
[ruffinelli2020] call the sLCWA
NegSamp
in their work.Initialize the training loop.
- Parameters
negative_sampler (
Union
[str
,NegativeSampler
,Type
[NegativeSampler
],None
]) – The class, instance, or name of the negative samplernegative_sampler_kwargs (
Optional
[Mapping
[str
,Any
]]) – Keyword arguments to pass to the negative sampler class on instantiation for every positive onekwargs – Additional keyword-based parameters passed to TrainingLoop.__init__