// Copyright (C) 2001 Jean-Marc Valin
#ifndef TRAINING_ALGO_H
#define TRAINING_ALGO_H
#include <vector>
namespace FD {
class FFNet;
class TrainingAlgo {
public:
};
class TrainingDeltaBarDelta : public TrainingAlgo {
public:
static void train(FFNet *net, std::vector<float *> tin, std::vector<float *> tout, int iter, float learnRate,
float increase, float decrease, int nbSets=1, bool rprop=false);
};
class TrainingSCG : public TrainingAlgo {
public:
static void train(FFNet *net, std::vector<float *> tin, std::vector<float *> tout, int iter, float sigma, float lambda);
};
class TrainingQProp : public TrainingAlgo {
public:
static void train(FFNet *net, std::vector<float *> tin, std::vector<float *> tout, int iter, float learnRate);
};
class TrainingWeightDeltaBarDelta : public TrainingAlgo {
public:
static void train(FFNet *net, std::vector<float *> tin, std::vector<float *> tout, std::vector<float *> learnWeights, int iter, float learnRate,
float increase, float decrease);
};
}//namespace FD
#endif
syntax highlighted by Code2HTML, v. 0.9.1