Bug fix.

izzat5233 · Dec 16, 2023 · f6e457e · f6e457e
1 parent a2211bb
commit f6e457e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/nn/network.h b/nn/network.h
@@ -12,8 +12,6 @@
 class nn::Network {
 private:
     const std::size_t size;
-
-private:
     vl_t layers;
     OutputLayer outputLayer;
     double alpha;

diff --git a/wasm/network_interface.h b/wasm/network_interface.h
@@ -148,7 +148,7 @@ class NetworkController {
     [[nodiscard]] std::vector<nn::vvd_t> getWeights() const {
         std::vector<nn::vvd_t> weights;
         weights.reserve(network->getSize());
-        for (std::size_t i = 0; i < weights.size(); ++i) {
+        for (std::size_t i = 0; i < network->getSize(); ++i) {
             nn::vvd_t layerWeights;
             layerWeights.reserve(network->get(i).size());
             for (auto &neuron: network->get(i)) {