org.deeplearning4j.optimize.listeners.callbacks

Class ModelSavingCallback

java.lang.Object

org.deeplearning4j.optimize.listeners.callbacks.ModelSavingCallback

All Implemented Interfaces:

EvaluationCallback

public class ModelSavingCallback
extends Object
implements EvaluationCallback

This callback will save model after each EvaluativeListener invocation. Filename template respects %d pattern, which will be replaced with integer value representing invocation number (not iteration!). I.e. if EvaluativeListener has frequency set to 50, it will be invoked once every 50 iterations, each invocation will increment number by 1. So, after 500 epochs there will be 10 invocations in total, and 10 models will be saved. PLEASE NOTE:

Author:

raver119@gmail.com

Field Summary

Fields

Modifier and Type	Field and Description
protected File	rootFolder
protected String	template

Constructor Summary

Constructors

Constructor and Description
ModelSavingCallback(File rootFolder, String fileNameTemplate) This constructor will create ModelSavingCallback instance that will save models in specified folder PLEASE NOTE: Make sure you have write access to the target folder
ModelSavingCallback(String fileNameTemplate) This constructor will create ModelSavingCallback instance that will save models in current folder PLEASE NOTE: Make sure you have write access to the current folder

Method Summary

Modifier and Type	Method and Description
void	call(EvaluativeListener listener, Model model, long invocationsCount, IEvaluation[] evaluations)
protected void	save(Model model, String filename) This method saves model

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

rootFolder

protected File rootFolder

template

protected String template

Constructor Detail

ModelSavingCallback

public ModelSavingCallback(@NonNull
                           String fileNameTemplate)

This constructor will create ModelSavingCallback instance that will save models in current folder PLEASE NOTE: Make sure you have write access to the current folder

Parameters:

fileNameTemplate -

ModelSavingCallback

public ModelSavingCallback(@NonNull
                           File rootFolder,
                           @NonNull
                           String fileNameTemplate)

This constructor will create ModelSavingCallback instance that will save models in specified folder PLEASE NOTE: Make sure you have write access to the target folder

Parameters:

rootFolder - File object referring to target folder

fileNameTemplate -

Method Detail

call

public void call(EvaluativeListener listener,
                 Model model,
                 long invocationsCount,
                 IEvaluation[] evaluations)

Specified by:

call in interface EvaluationCallback

save

protected void save(Model model,
                    String filename)

This method saves model

Parameters:

model -

filename -