First implementation of weight updates. Very slow rate of change in the output value.
This commit is contained in:
parent
de06daaad3
commit
a79abb5db1
5 changed files with 43 additions and 12 deletions
17
Layer.cpp
17
Layer.cpp
|
@ -51,6 +51,21 @@ void Layer::connectTo(const Layer & nextLayer)
|
|||
}
|
||||
}
|
||||
|
||||
void Layer::updateInputWeights(const Layer & prevLayer)
|
||||
void Layer::updateInputWeights(Layer & prevLayer)
|
||||
{
|
||||
static const double trainingRate = 0.8;
|
||||
|
||||
for (size_t currentLayerIndex = 0; currentLayerIndex < size() - 1; ++currentLayerIndex)
|
||||
{
|
||||
Neuron &targetNeuron = at(currentLayerIndex);
|
||||
|
||||
for (size_t prevLayerIndex = 0; prevLayerIndex < prevLayer.size(); ++prevLayerIndex)
|
||||
{
|
||||
Neuron &sourceNeuron = prevLayer.at(prevLayerIndex);
|
||||
|
||||
sourceNeuron.setOutputWeight(currentLayerIndex,
|
||||
sourceNeuron.getOutputWeight(currentLayerIndex) +
|
||||
sourceNeuron.getOutputValue() * targetNeuron.getGradient() * trainingRate);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
2
Layer.h
2
Layer.h
|
@ -14,5 +14,5 @@ public:
|
|||
double getWeightedSum(int outputNeuron) const;
|
||||
void connectTo(const Layer & nextLayer);
|
||||
|
||||
void updateInputWeights(const Layer &prevLayer);
|
||||
void updateInputWeights(Layer &prevLayer);
|
||||
};
|
||||
|
|
|
@ -14,6 +14,8 @@ int main()
|
|||
|
||||
Net myNet({ inputValues.size(), 4, targetValues.size() });
|
||||
|
||||
for (int i = 0; i < 20; ++i)
|
||||
{
|
||||
myNet.feedForward(inputValues);
|
||||
|
||||
std::vector<double> outputValues = myNet.getOutput();
|
||||
|
@ -27,6 +29,7 @@ int main()
|
|||
|
||||
myNet.backProp(targetValues);
|
||||
}
|
||||
}
|
||||
catch (std::exception &ex)
|
||||
{
|
||||
std::cerr << "Error: " << ex.what() << std::endl;
|
||||
|
|
10
Neuron.cpp
10
Neuron.cpp
|
@ -84,3 +84,13 @@ double Neuron::getGradient() const
|
|||
return gradient;
|
||||
}
|
||||
|
||||
double Neuron::getOutputWeight(size_t index) const
|
||||
{
|
||||
return outputWeights.at(index);
|
||||
}
|
||||
|
||||
void Neuron::setOutputWeight(size_t index, double value)
|
||||
{
|
||||
outputWeights.at(index) = value;
|
||||
}
|
||||
|
||||
|
|
3
Neuron.h
3
Neuron.h
|
@ -25,6 +25,9 @@ public:
|
|||
|
||||
double getGradient() const;
|
||||
|
||||
double getOutputWeight(size_t index) const;
|
||||
void setOutputWeight(size_t index, double value);
|
||||
|
||||
private:
|
||||
static double transferFunction(double inputValue);
|
||||
static double transferFunctionDerivative(double inputValue);
|
||||
|
|
Loading…
Reference in a new issue