Interface EarlyStoppingModelSaver<T extends Model>

    • Method Summary

      All Methods Instance Methods Abstract Methods 
      Modifier and Type Method Description
      T getBestModel()
      Retrieve the best model that was previously saved
      T getLatestModel()
      Retrieve the most recent model that was previously saved
      void saveBestModel​(T net, double score)
      Save the best model (so far) learned during early stopping training
      void saveLatestModel​(T net, double score)
      Save the latest (most recent) model learned during early stopping
    • Method Detail

      • saveBestModel

        void saveBestModel​(T net,
                           double score)
                    throws IOException
        Save the best model (so far) learned during early stopping training
        Throws:
        IOException
      • saveLatestModel

        void saveLatestModel​(T net,
                             double score)
                      throws IOException
        Save the latest (most recent) model learned during early stopping
        Throws:
        IOException
      • getBestModel

        T getBestModel()
                throws IOException
        Retrieve the best model that was previously saved
        Throws:
        IOException
      • getLatestModel

        T getLatestModel()
                  throws IOException
        Retrieve the most recent model that was previously saved
        Throws:
        IOException