[ Fix ] Fix Logistic Regression Example Error
authorjijoong.moon <jijoong.moon@samsung.com>
Thu, 14 Jan 2021 03:58:31 +0000 (12:58 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 21 Jan 2021 01:59:47 +0000 (10:59 +0900)
This PR includes fixes about logistic regression application

Change forwarding function

**Self evaluation:**
1. Build test:  [X]Passed [ ]Failed [ ]Skipped
2. Run test:  [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: jijoong.moon <jijoong.moon@samsung.com>
Applications/LogisticRegression/jni/main.cpp

index 4df7997..d1ae924 100644 (file)
@@ -217,7 +217,8 @@ int main(int argc, char *argv[]) {
 
       try {
         float answer =
-          NN.forwarding({MAKE_SHARED_TENSOR(nntrainer::Tensor({o}))})[0]
+          NN.forwarding({MAKE_SHARED_TENSOR(nntrainer::Tensor({o}))},
+                        {MAKE_SHARED_TENSOR(nntrainer::Tensor({l}))})[0]
             ->apply(stepFunction)
             .getValue(0, 0, 0, 0);
         std::cout << answer << " : " << l[0] << std::endl;