ml: fix caching of internal state when changing the impl in KNearest

9b0ef7bb · berak · 7b36e575 · 9b0ef7bb
Commit 9b0ef7bb authored May 11, 2018 by berak
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

knearest.cpp modules/ml/src/knearest.cpp +9 -0

No files found.
--- a/modules/ml/src/knearest.cpp
+++ b/modules/ml/src/knearest.cpp
@@ -437,7 +437,16 @@ public:
    {
        if (val != BRUTE_FORCE && val != KDTREE)
            val = BRUTE_FORCE;
+
+        int k = getDefaultK();
+        int e = getEmax();
+        bool c = getIsClassifier();
+
        initImpl(val);
+
+        setDefaultK(k);
+        setEmax(e);
+        setIsClassifier(c);
    }

 public: