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

Modifier and Type	Method and Description
int	getBatchSize() Gets the batch size
int	getNumEpochs() Gets the number of epochs to train the model
java.lang.Long	getSeed() Gets the random seed
boolean	isShuffle() Gets whether the model will shuffle or not
void	setLoss(java.lang.Double loss) Sets the loss after training

Method Detail

getBatchSize

int getBatchSize()

Gets the batch size

Returns:

batch size

Since:

23.2

getNumEpochs

int getNumEpochs()

Gets the number of epochs to train the model

Returns:

number of epochs to train the model

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