fix an error when loading double precision model file

Xtra-Computing · Feb 20, 2019 · 6062b4f · 6062b4f
1 parent 7b6b743
commit 6062b4f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/thundersvm/model/svmmodel.cpp b/src/thundersvm/model/svmmodel.cpp
@@ -276,7 +276,12 @@ void SvmModel::load_from_file(string path) {
                 string tuple;
                 while (ss >> tuple) {
                     sv.back().emplace_back(0, 0);
+//todo use streaming operator >> here to avoid ifdef
+#ifdef USE_DOUBLE
+                    CHECK_EQ(sscanf(tuple.c_str(), "%d:%lf", &sv.back().back().index, &sv.back().back().value), 2)
+#else
                     CHECK_EQ(sscanf(tuple.c_str(), "%d:%f", &sv.back().back().index, &sv.back().back().value), 2)
+#endif
                         << "error when loading model file";
                 };
             }
@@ -368,4 +373,4 @@ void SvmModel::get_param(char* kernel_type, int* degree, float* gamma, float* co
     *gamma = param.gamma;
     *coef0 = param.coef0;
     *probability = param.probability;
-}
+}