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

Hyperparameter changes in early stopping

parent e5b14d29
Branches master
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, frequency=5) evaluation_triples_factory= eval_data, frequency=5, relative_delat=0.0, metric='arithmetic_mean_rank', larger_is_better=False)
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