Added logistic regression check output

Signed-off-by: Alejandro Saucedo <[email protected]>
SlopeCraft · Nov 26, 2022 · 342afcc · 342afcc
1 parent cf1dec9
commit 342afcc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/examples/logistic_regression/src/main.cpp b/examples/logistic_regression/src/main.cpp
@@ -63,4 +63,10 @@ main()
     std::cout << "w1: " << wIn->data()[0] << std::endl;
     std::cout << "w2: " << wIn->data()[1] << std::endl;
     std::cout << "b: " << bIn->data()[0] << std::endl;
-}
+
+    if (wIn->data()[0] > 0.0 ||
+            wIn->data()[1] < 1.0 ||
+            bIn->data()[0] > 0.0) {
+        throw std::runtime_error("Result does not match");
+    }
+}