DeepCL  v4.0.0rc5
Deep convolutional neural networks using OpenCL
 All Classes Functions
Public Member Functions | List of all members
NetLearner Class Reference

Runs multiple learning epochs using Batcher objects. More...

Inherits NetLearnerBase.

Public Member Functions

publicapi NetLearner (Trainable *net, int Ntrain, float *trainData, int *trainLabels, int Ntest, float *testData, int *testLabels, int batchSize)
 
publicapi VIRTUAL void reset ()
 
publicapi VIRTUAL bool tickBatch ()
 
publicapi VIRTUAL bool getEpochDone ()
 
publicapi VIRTUAL int getNextEpoch ()
 
publicapi VIRTUAL int getNextBatch ()
 
publicapi VIRTUAL int getNTrain ()
 
publicapi VIRTUAL int getBatchNumRight ()
 
publicapi VIRTUAL float getBatchLoss ()
 
publicapi VIRTUAL bool tickEpoch ()
 
publicapi VIRTUAL void run ()
 
publicapi VIRTUAL bool isLearningDone ()
 
publicapi VIRTUAL void setLearningRate (float learningRate)
 
publicapi VIRTUAL void learn (float learningRate)
 

Detailed Description

Runs multiple learning epochs using Batcher objects.

Handles learning the neural net, ie running multiple epochs. Uses two Batchers, one for training, one for testing, to learn the epochs.

This class expects the data to be already in memory. If the data is really big, wont fit in memory, you probably want to use something more like NetLearnerOnDemand, which can load in a chunk of data from datafiles at a time

Public API

Constructor & Destructor Documentation

publicapi NetLearner::NetLearner ( Trainable *  net,
int  Ntrain,
float *  trainData,
int *  trainLabels,
int  Ntest,
float *  testData,
int *  testLabels,
int  batchSize 
)
Public API

Member Function Documentation

publicapi VIRTUAL void NetLearner::reset ( )
Public API
publicapi VIRTUAL bool NetLearner::tickBatch ( )
Public API
publicapi VIRTUAL bool NetLearner::getEpochDone ( )
Public API
publicapi VIRTUAL int NetLearner::getNextEpoch ( )
Public API
publicapi VIRTUAL int NetLearner::getNextBatch ( )
Public API
publicapi VIRTUAL int NetLearner::getNTrain ( )
Public API
publicapi VIRTUAL int NetLearner::getBatchNumRight ( )
Public API
publicapi VIRTUAL float NetLearner::getBatchLoss ( )
Public API
publicapi VIRTUAL bool NetLearner::tickEpoch ( )
Public API
publicapi VIRTUAL void NetLearner::run ( )
Public API
publicapi VIRTUAL bool NetLearner::isLearningDone ( )
Public API
publicapi VIRTUAL void NetLearner::setLearningRate ( float  learningRate)
Public API
publicapi VIRTUAL void NetLearner::learn ( float  learningRate)
Public API

The documentation for this class was generated from the following files: