停止器
早停器。
以下代码将创建一个场景,在该场景中,当在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),
... )
类
|
用于停止训练的装置。 |
|
一个什么都不做的停止器。 |
|
用于早期停止的装置。 |
变量
一个用于停止器的解析器 |
类继承图
