shin/NeuralNetworkLib
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

modified BP interface

Browse Source
This commit is contained in:
Shin
2016-02-24 19:05:26 +01:00
parent 47de0fa08b
commit c45f12f53c
10 changed files with 195 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
include/NeuralNetwork/Learning/BackPropagation.h
View File

@@ -2,7 +2,9 @@
#include <vector>
#include <cmath>
#include <NeuralNetwork/FeedForward/Network.h>
#include "CorrectionFunction/Linear.h"
namespace NeuralNetwork {
namespace Learning {
@@ -13,21 +15,23 @@ namespace Learning {
class BackPropagation {
public:
inline BackPropagation(FeedForward::Network &feedForwardNetwork): network(feedForwardNetwork), learningCoefficient(0.4), deltas() {
inline BackPropagation(FeedForward::Network &feedForwardNetwork, CorrectionFunction::CorrectionFunction *correction = new CorrectionFunction::Linear()):
network(feedForwardNetwork), correctionFunction(correction),learningCoefficient(0.4), deltas() {
resize();
}
virtual ~BackPropagation() {
delete correctionFunction;
}
BackPropagation(const BackPropagation&)=delete;
BackPropagation& operator=(const NeuralNetwork::Learning::BackPropagation&) = delete;
void teach(const std::vector<float> &input, const std::vector<float> &output);
inline virtual void setLearningCoefficient (const float& coefficient) { learningCoefficient=coefficient; }
protected:
inline virtual float correction(const float & expected, const float &computed) const {
return expected-computed;
};
inline void resize() {
if(deltas.size()!=network.size())
@@ -41,6 +45,8 @@ namespace Learning {
FeedForward::Network &network;
CorrectionFunction::CorrectionFunction *correctionFunction;
float learningCoefficient;
std::vector<std::vector<float>> deltas;