From 13a1804dda731e45541a5b2ef05a0a9775d9904e Mon Sep 17 00:00:00 2001
From: Mikhalkovich Stanislav <miks@math.sfedu.ru>
Date: Wed, 15 Apr 2026 22:21:22 +0300
Subject: [PATCH] =?UTF-8?q?ML=20-=20=D0=B8=D1=81=D0=BF=D1=80=D0=B0=D0=B2?=
 =?UTF-8?q?=D0=BB=D0=B5=D0=BD=D0=B8=D0=B5=20=D0=BF=D0=B0=D0=B4=D0=B5=D0=BD?=
 =?UTF-8?q?=D0=B8=D1=8F=20=D0=B2=20DTC=20-=20=D0=B8=D0=B7=20=D0=B7=D0=B0?=
 =?UTF-8?q?=20=D1=80=D0=B0=D0=BD=D0=BD=D0=B5=D0=B9=20=D0=B8=D0=BD=D0=B8?=
 =?UTF-8?q?=D1=86=D0=B8=D0=B0=D0=BB=D0=B8=D0=B7=D0=B0=D1=86=D0=B8=D0=B8=20?=
 =?UTF-8?q?=D0=BA=D1=80=D0=B8=D1=82=D0=B5=D1=80=D0=B8=D1=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bin/Lib/MLModelsABC.pas | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/bin/Lib/MLModelsABC.pas b/bin/Lib/MLModelsABC.pas
index cb0c8aab3..010c1b106 100644
--- a/bin/Lib/MLModelsABC.pas
+++ b/bin/Lib/MLModelsABC.pas
@@ -4391,7 +4391,8 @@ begin
   fMaxDepth := maxDepth;
   fMinSamplesSplit := minSamplesSplit;
   fMinSamplesLeaf := minSamplesLeaf;
-  fCriterion := if criterion = nil then new GiniCriterion else criterion;
+  
+  fCriterion := criterion;
 
   fRandomSeed := ResolveRandomSeed(seed, fUserProvidedSeed);
   fRng := new System.Random(fRandomSeed);
@@ -4437,6 +4438,9 @@ begin
   // --- encode
   var classes: array of integer;
   var yEncArr := EncodeLabelsInt(yInt, classes);
+  
+  if fCriterion = nil then
+    fCriterion := new GiniCriterion(classes.Length);
 
   if classes.Length < 2 then
     ArgumentError(ER_NEED_AT_LEAST_TWO_CLASSES);