fix: svm prediction with alll db combinations

jolibrain · Sep 22, 2020 · d8e2961 · d8e2961
1 parent 6e925f2
commit d8e2961
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/backends/caffe/caffeinputconns.cc b/src/backends/caffe/caffeinputconns.cc
@@ -2466,7 +2466,7 @@ namespace dd
       const int &label, const std::unordered_map<int, double> &vals,
       const int &count)
   {
-    if (!_db)
+    if (!_db || !_train)
       {
         SVMInputFileConn::add_test_svmline(label, vals, count);
         return;

diff --git a/src/svminputfileconn.cc b/src/svminputfileconn.cc
@@ -168,7 +168,7 @@ namespace dd
         std::unordered_map<int, double> vals;
         int label;
         read_svm_line(hline, vals, label);
-        if (train_lines > 0 && nlines < train_lines)
+        if (train_lines == 0 || (train_lines > 0 && nlines < train_lines))
           add_train_svmline(label, vals, nlines);
         else
           {