From 9b0ef7bb17dc0e5f3b83bd9dc11e1756ce452abb Mon Sep 17 00:00:00 2001 From: berak Date: Fri, 11 May 2018 11:47:02 +0200 Subject: [PATCH] ml: fix caching of internal state when changing the impl in KNearest --- modules/ml/src/knearest.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/modules/ml/src/knearest.cpp b/modules/ml/src/knearest.cpp index 3bf5c6ad28..30a0951bf3 100644 --- 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: