LCWALit模块

class LCWALitModule(dataset: str | Dataset | type[Dataset] | None = 'nations', dataset_kwargs: Mapping[str, Any] | None = None, mode: Literal['training', 'validation', 'testing'] | None = None, model: str | Model | type[Model] | None = 'distmult', model_kwargs: Mapping[str, Any] | None = None, batch_size: int = 32, learning_rate: float = 0.001, label_smoothing: float = 0.0, optimizer: str | Optimizer | type[Optimizer] | None = None, optimizer_kwargs: Mapping[str, Any] | None = None)

基础类：LitModule

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

另请参阅

https://github.com/pykeen/pykeen/pull/905

创建闪电模块。

Parameters:

• dataset (str | Dataset | type[Dataset] | None) – 数据集或其提示

• dataset_kwargs (Mapping[str, Any] | None) – 传递给数据集的额外基于关键字的参数

• mode (Literal['training', 'validation', 'testing'] | None) – 归纳模式；默认为传导训练

• model (str | Model | type[Model] | None) – 模型或其提示

• model_kwargs (Mapping[str, Any] | None) – 传递给模型的额外基于关键字的参数

• batch_size (int) – 训练批次大小

• learning_rate (float) – 学习率

• label_smoothing (float) – 标签平滑

• optimizer (str | Optimizer | type[Optimizer] | None) – 优化器，或其提示

• optimizer_kwargs (Mapping[str, Any] | None) – 传递给优化器的额外基于关键字的参数。不应包含 lr 或 params。