SLCWALitModule

class SLCWALitModule(*, negative_sampler=None, negative_sampler_kwargs=None, **kwargs)

Bases: LitModule

A PyTorch Lightning module for training a model with sLCWA training loop.

Initialize the lightning module.

Parameters:

• negative_sampler (Union[str, NegativeSampler, Type[NegativeSampler], None]) – the negative sampler, cf. pykeen.triples.CoreTriplesFactory.create_slcwa_instances()

• negative_sampler_kwargs (Optional[Mapping[str, Any]]) – keyword-based parameters passed to the negative sampler, cf. pykeen.triples.CoreTriplesFactory.create_slcwa_instances()

• kwargs – additional keyword-based parameters passed to LitModule.__init__()