Package 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

Field Summary

Fields

Modifier and Type	Field	Description
protected File	rootFolder
protected String	template

Constructor Summary

Constructors

Constructor	Description
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
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

Method Summary

Modifier and Type	Method	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
                           @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
                           @NonNull File rootFolder,
                           @NonNull
                           @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 -