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

7 years ago · 9b0ef7bb17
parent 7b36e57551
commit 9b0ef7bb17
1 changed files with 9 additions and 0 deletions
--- 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: