diff options
Diffstat (limited to 'src/brain')
-rw-r--r-- | src/brain/neural_link.cpp | 4 | ||||
-rw-r--r-- | src/brain/neural_network.cpp | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/brain/neural_link.cpp b/src/brain/neural_link.cpp index c61feb4..6997bd6 100644 --- a/src/brain/neural_link.cpp +++ b/src/brain/neural_link.cpp @@ -1,11 +1,11 @@ #include "brain/neural_link.h" -NeuralLink::NeuralLink(std::weak_ptr<NeuralNode> input, double weight) +NeuralLink::NeuralLink(NeuralNode* input, double weight) :_input(input), _weight(weight) { } double NeuralLink::weightedActivation() const { - return _input.lock()->activation()*_weight; + return _input->activation()*_weight; } diff --git a/src/brain/neural_network.cpp b/src/brain/neural_network.cpp index 64d6979..bacdbed 100644 --- a/src/brain/neural_network.cpp +++ b/src/brain/neural_network.cpp @@ -84,7 +84,7 @@ void NeuralNetwork::parseFile(std::istream &&file) void NeuralNetwork::addLink(std::shared_ptr<NeuralNode> source, std::shared_ptr<Neuron> destination, double weight) { - NeuralLink link(source, weight); + NeuralLink link(source.get(), weight); destination->addInput(std::move(link)); } |