Interface BaseModelConfig

    • 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