aselimov/neural_net
1
0
Fork 0
generated from aselimov/cpp_project_template
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix feed forward implementation and tests

Browse Source
This commit is contained in:
Alex Selimov 2025-03-29 12:52:35 -04:00
parent 59d27f47f6
commit 7cefc9baf1
2 changed files with 20 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/utility.hpp
View File

@ -16,12 +16,11 @@ std::vector<float> feed_layer(std::vector<float>::iterator weight_start,
// Calculate the new A vector from the current weights // Calculate the new A vector from the current weights
std::vector<float> Anew; std::vector<float> Anew;
Anew.reserve(std::distance(weight_start, weight_end)); Anew.reserve(std::distance(weight_start, weight_end));
std::transform(weight_start, weight_end, Anew.begin(), std::transform(
[&A, &activation_func](float weight) { weight_start, weight_end, std::back_inserter(Anew), [&A](float weight) {
float summed_weight = std::accumulate( float summed_weight = std::accumulate(
A.begin(), A.end(), 0.0f, [&weight](float acc, float a) { A.begin(), A.end(), 0.0f,
return acc + a * weight; [&weight](float acc, float a) { return acc + a * weight; });
});
return summed_weight; return summed_weight;
}); });
activation_func(Anew); activation_func(Anew);

14
tests/unit_tests/test_utility.cpp
View File

@ -7,6 +7,9 @@
struct Identity { struct Identity {
void operator()(std::vector<float> &x) const { void operator()(std::vector<float> &x) const {
// Identity function - no change to values // Identity function - no change to values
// Below statement is needed to remove compiler warning about unused var
// warning
(void)x;
} }
}; };
@ -14,9 +17,6 @@ TEST(UtilityTest, FeedLayerIdentityTest) {
// Test with identity activation function for simple verification // Test with identity activation function for simple verification
// Input: [1, 2] // Input: [1, 2]
// Weights: [0.5, -0.5, 1.0, -1.0] // Weights: [0.5, -0.5, 1.0, -1.0]
// Expected: [0.5, -1.0] (manually calculated)
// First output: 1.0 * 0.5 + 2.0 * -0.5 = 0.5
// Second output: 1.0 * 1.0 + 2.0 * -1.0 = -1.0
std::vector<float> weights = {0.5, -0.5, 1.0, -1.0}; std::vector<float> weights = {0.5, -0.5, 1.0, -1.0};
std::vector<float> input = {1.0, 2.0}; std::vector<float> input = {1.0, 2.0};
@ -25,9 +25,11 @@ TEST(UtilityTest, FeedLayerIdentityTest) {
auto output = Utilities::feed_layer<Identity>(weights.begin(), weights.end(), auto output = Utilities::feed_layer<Identity>(weights.begin(), weights.end(),
input, identity); input, identity);
ASSERT_EQ(output.size(), 2); ASSERT_EQ(output.size(), 4);
EXPECT_NEAR(output[0], 0.5f, 1e-5); // 1.0 * 0.5 + 2.0 * -0.5 EXPECT_NEAR(output[0], 1.5f, 1e-5); // 1.0 * 0.5 + 2.0 * 0.5
EXPECT_NEAR(output[1], -1.0f, 1e-5); // 1.0 * 1.0 + 2.0 * -1.0 EXPECT_NEAR(output[1], -1.5f, 1e-5); // 1.0 * -0.5 + 2.0 * -0.5
EXPECT_NEAR(output[2], 3.0f, 1e-5); // 1.0 * 1.0 + 2.0 * 1.0
EXPECT_NEAR(output[3], -3.0f, 1e-5); // 1.0 * -1.0 + 2.0 * -1.0
} }
TEST(UtilityTest, FeedLayerSigmoidTest) { TEST(UtilityTest, FeedLayerSigmoidTest) {