public interface HpOptimizer
Hyperparameter
s.Hyperparameter
Modifier and Type | Method and Description |
---|---|
ai.djl.util.Pair<HpSet,java.lang.Float> |
getBest()
Returns the best hyperparameters and loss.
|
float |
getLoss(HpSet config)
Returns the recorded loss.
|
HpSet |
nextConfig()
Returns the next hyperparameters to test.
|
void |
update(HpSet config,
float loss)
Updates the optimizer with the results of a hyperparameter test.
|
HpSet nextConfig()
void update(HpSet config, float loss)
config
- the tested hyperparametersloss
- the validation loss from training with those hyperparametersfloat getLoss(HpSet config)
config
- the hyperparameters that were trained withjava.util.NoSuchElementException
- if the hyperparameters were not trained with beforeai.djl.util.Pair<HpSet,java.lang.Float> getBest()