SLCWALitModule

class SLCWALitModule(*, negative_sampler: str | NegativeSampler | type[NegativeSampler] | None = None, negative_sampler_kwargs: Mapping[str, Any] | None = None, **kwargs)[source]

基础类：LitModule

一个用于使用sLCWA训练循环训练模型的PyTorch Lightning模块。

初始化闪电模块。

Parameters:

negative_sampler (str | NegativeSampler | type[NegativeSampler] | None) – 负采样器，参见 pykeen.triples.CoreTriplesFactory.create_slcwa_instances()
negative_sampler_kwargs (Mapping[str, Any] | None) – 传递给负采样器的基于关键字的参数，参见 pykeen.triples.CoreTriplesFactory.create_slcwa_instances()
kwargs – 传递给 LitModule.__init__() 的额外基于关键字的参数