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

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

  • Method Details

    • 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

      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(Double loss)
      Sets the loss after training
      Parameters:
      loss - the value of the training loss
      Since:
      23.2