Package oracle.pgx.config.mllib
Interface BaseModelConfig
-
- All Known Implementing Classes:
DeepWalkModelConfig
,Pg2vecModelConfig
public interface BaseModelConfig
Abstract base configuration class for all models- Since:
- 23.2
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description int
getBatchSize()
Gets the batch sizeint
getNumEpochs()
Gets the number of epochs to train the modeljava.lang.Long
getSeed()
Gets the random seedboolean
isShuffle()
Gets whether the model will shuffle or notvoid
setLoss(java.lang.Double loss)
Sets the loss after training
-
-
-
Method Detail
-
getNumEpochs
int getNumEpochs()
Gets the number of epochs to train the model- Returns:
- number of epochs to train the model
- Since:
- 23.2
-
getBatchSize
int getBatchSize()
Gets the batch size- Returns:
- batch size
- Since:
- 23.2
-
getSeed
java.lang.Long getSeed()
Gets the random seed- Returns:
- random seed
- Since:
- 23.2
-
isShuffle
boolean isShuffle()
Gets whether the model will shuffle or not- Returns:
- whether the model will shuffle or not
- Since:
- 23.2
-
setLoss
void setLoss(java.lang.Double loss)
Sets the loss after training- Parameters:
loss
- the value of the training loss- Since:
- 23.2
-
-