Merge pull request #1612 from shimat/fix_RTreesActiveVarCount

RTrees.ActiveVarCount: bool -> int
shimat · Sep 30, 2023 · 1b75869 · 1b75869
2 parents b53a92e + b51c3a1
commit 1b75869
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/OpenCvSharp/Modules/ml/RTrees.cs b/src/OpenCvSharp/Modules/ml/RTrees.cs
@@ -101,21 +101,21 @@ public bool CalculateVarImportance
     /// The size of the randomly selected subset of features at each tree node 
     /// and that are used to find the best split(s).
     /// </summary>
-    public bool ActiveVarCount
+    public int ActiveVarCount
     {
         get
         {
             ThrowIfDisposed();
             NativeMethods.HandleException(
                 NativeMethods.ml_RTrees_getActiveVarCount(ptr, out var ret));
             GC.KeepAlive(this);
-            return ret != 0;
+            return ret;
         }
         set
         {
             ThrowIfDisposed();
             NativeMethods.HandleException(
-                NativeMethods.ml_RTrees_setActiveVarCount(ptr, value ? 1 : 0));
+                NativeMethods.ml_RTrees_setActiveVarCount(ptr, value));
             GC.KeepAlive(this);
         }
     }