停止器

早停器。

以下代码将创建一个场景,在该场景中,当在pykeen.datasets.Nations数据集上训练pykeen.models.TransE时,训练将(相当)早地停止。

>>> from pykeen.pipeline import pipeline
>>> pipeline_result = pipeline(
...     dataset='nations',
...     model='transe',
...     model_kwargs=dict(embedding_dim=20, scoring_fct_norm=1),
...     optimizer='SGD',
...     optimizer_kwargs=dict(lr=0.01),
...     loss='marginranking',
...     loss_kwargs=dict(margin=1),
...     training_loop='slcwa',
...     training_kwargs=dict(num_epochs=100, batch_size=128),
...     negative_sampler='basic',
...     negative_sampler_kwargs=dict(num_negs_per_pos=1),
...     evaluator_kwargs=dict(filtered=True),
...     evaluation_kwargs=dict(batch_size=128),
...     stopper='early',
...     stopper_kwargs=dict(frequency=5, patience=2, relative_delta=0.002),
... )

Stopper(*args, **kwargs)

用于停止训练的装置。

NopStopper(*args, **kwargs)

一个什么都不做的停止器。

EarlyStopper(model, evaluator, ...)

用于早期停止的装置。

变量

stopper_resolver

一个用于停止器的解析器

类继承图

Inheritance diagram of pykeen.stoppers.stopper.Stopper, pykeen.stoppers.stopper.NopStopper, pykeen.stoppers.early_stopping.EarlyStopper