class StochasticGradient

Trainer for GradientMachine.

Inheritance:

Public Methods

StochasticGradient(GradientMachine* machine_, Criterion* criterion_)

Take the machine_ to train, the train data_, the criterion_ to use, and the optimizer_ to use

Inherited from Trainer:

Public Methods

virtual void train(DataSet* data_, MeasurerList* measurers)

virtual void test(MeasurerList* measurers)

static Allocator* extractMeasurers(MeasurerList* measurers, DataSet* train, DataSet*** datas, Measurer**** meas, int** n_meas, int* n_datas)

virtual void loadXFile(XFile* file)

virtual void saveXFile(XFile* file)

Inherited from Object:

Public Fields

Allocator* allocator

Public Methods

void addOption(const char* name, int size, void* ptr, const char* help="")

void addIOption(const char* name, int* ptr, int init_value, const char* help="")

void addROption(const char* name, real* ptr, real init_value, const char* help="")

void addBOption(const char* name, bool* ptr, bool init_value, const char* help="")

void addOOption(const char* name, Object** ptr, Object* init_value, const char* help="")

void setOption(const char* name, void* ptr)

void setIOption(const char* name, int option)

void setROption(const char* name, real option)

void setBOption(const char* name, bool option)

void setOOption(const char* name, Object* option)

void load(const char* filename)

void save(const char* filename)

void* operator new(size_t size, Allocator* allocator_=NULL)

void* operator new(size_t size, Allocator* allocator_, void* ptr_)

void operator delete(void* ptr)

Documentation

Trainer for GradientMachine. Given a machine and a criterion, train the machine using a stochastic gradient descent.

Options:

"end accuracy"	real	end accuracy	[0.0001]
"learning rate"	real	learning rate	[0.01]
"learning rate decay"	real	learning rate decay	[0]
"max iter"	int	maximum number of iterations	[-1]
"shuffle"	bool	shuffle the dataset	[true]

StochasticGradient(GradientMachine* machine_, Criterion* criterion_)

Take the machine_ to train, the train data_, the criterion_ to use, and the optimizer_ to use

This class has no child classes.

Author:

Ronan Collobert (collober@idiap.ch)

Alphabetic index HTML hierarchy of classes or Java