Skip to content
Snippets Groups Projects
Commit e5b14d29 authored by lenhy's avatar lenhy
Browse files

More frequent checks for early stopping

parent 072a9008
No related branches found
No related tags found
No related merge requests found
...@@ -44,7 +44,7 @@ def get_model(name: str, triples, inverse_triples, regularizer): ...@@ -44,7 +44,7 @@ def get_model(name: str, triples, inverse_triples, regularizer):
def train(model, train_data, eval_data, optimizer, evaluator, n_epochs: int = 20, batch_size: int = 256): def train(model, train_data, eval_data, optimizer, evaluator, n_epochs: int = 20, batch_size: int = 256):
early_stopper = EarlyStopper(model=model, evaluator= evaluator, training_triples_factory=train_data, early_stopper = EarlyStopper(model=model, evaluator= evaluator, training_triples_factory=train_data,
evaluation_triples_factory= eval_data) evaluation_triples_factory= eval_data, frequency=5)
training_loop = SLCWATrainingLoop( training_loop = SLCWATrainingLoop(
model=model, model=model,
triples_factory=train_data, triples_factory=train_data,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment