public class EarlyStoppingGraphTrainer extends BaseEarlyStoppingTrainer<ComputationGraph>
ComputationGraph
esConfig, model
Constructor and Description |
---|
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
DataSetIterator |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
MultiDataSetIterator |
Modifier and Type | Method and Description |
---|---|
protected void |
fit(org.nd4j.linalg.dataset.DataSet ds) |
protected void |
fit(org.nd4j.linalg.dataset.api.MultiDataSet mds) |
fit, reset, setListener
public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train)
esConfig
- Configurationnet
- Network to train using early stoppingtrain
- DataSetIterator for training the networkpublic EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, org.nd4j.linalg.dataset.api.iterator.DataSetIterator train, EarlyStoppingListener<ComputationGraph> listener)
DataSetIterator
esConfig
- Configurationnet
- Network to train using early stoppingtrain
- DataSetIterator for training the networklistener
- Early stopping listener. May be null.public EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig, ComputationGraph net, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator train, EarlyStoppingListener<ComputationGraph> listener)
MultiDataSetIterator
esConfig
- Configurationnet
- Network to train using early stoppingtrain
- DataSetIterator for training the networklistener
- Early stopping listener. May be null.protected void fit(org.nd4j.linalg.dataset.DataSet ds)
fit
in class BaseEarlyStoppingTrainer<ComputationGraph>
protected void fit(org.nd4j.linalg.dataset.api.MultiDataSet mds)
fit
in class BaseEarlyStoppingTrainer<ComputationGraph>