test_class.declarator: partsToBeTested = new int[] {1, 2, 3, 4, 5, 7}
test_class.modifier: private static final
test_class.original_string: private static final int[] partsToBeTested = new int[] {1, 2, 3, 4, 5, 7};
test_class.type: int[]
test_class.var_name: partsToBeTested
test_class.declarator: parts
test_class.modifier: @Parameterized.Parameter()
      public
test_class.original_string: @Parameterized.Parameter()
      public int parts;
test_class.type: int
test_class.var_name: parts
test_class.declarator: useIdx
test_class.modifier: @Parameterized.Parameter(1)
      public
test_class.original_string: @Parameterized.Parameter(1)
      public int useIdx;
test_class.type: int
test_class.var_name: useIdx
test_class.file: modules/ml/src/test/java/org/apache/ignite/ml/tree/DecisionTreeClassificationTrainerTest.java
test_class.identifier: DecisionTreeClassificationTrainerTest
test_case.body: @Test
      public void testFit() {
          int size = 100;
  
          Map<Integer, double[]> data = new HashMap<>();
  
          Random rnd = new Random(0);
          for (int i = 0; i < size; i++) {
              double x = rnd.nextDouble() - 0.5;
              data.put(i, new double[] {x, x > 0 ? 1 : 0});
          }
  
          DecisionTreeClassificationTrainer trainer = new DecisionTreeClassificationTrainer(1, 0)
              .withUseIndex(useIdx == 1);
  
          DecisionTreeNode tree = trainer.fit(data, parts, new DoubleArrayVectorizer<Integer>().labeled(1));
  
          assertTrue(tree instanceof DecisionTreeConditionalNode);
  
          DecisionTreeConditionalNode node = (DecisionTreeConditionalNode)tree;
  
          assertEquals(0, node.getThreshold(), 1e-3);
          assertEquals(0, node.getCol());
          assertNotNull(node.toString());
          assertNotNull(node.toString(true));
          assertNotNull(node.toString(false));
  
          assertTrue(node.getThenNode() instanceof DecisionTreeLeafNode);
          assertTrue(node.getElseNode() instanceof DecisionTreeLeafNode);
  
          DecisionTreeLeafNode thenNode = (DecisionTreeLeafNode)node.getThenNode();
          DecisionTreeLeafNode elseNode = (DecisionTreeLeafNode)node.getElseNode();
  
          assertEquals(1, thenNode.getVal(), 1e-10);
          assertEquals(0, elseNode.getVal(), 1e-10);
  
          assertNotNull(thenNode.toString());
          assertNotNull(thenNode.toString(true));
          assertNotNull(thenNode.toString(false));
      }
test_case.class_method_signature: DecisionTreeClassificationTrainerTest.testFit()
test_case.full_signature: @Test public void testFit()
test_case.identifier: testFit
test_case.invocations: nextDouble put withUseIndex fit labeled assertTrue assertEquals getThreshold assertEquals getCol assertNotNull toString assertNotNull toString assertNotNull toString assertTrue getThenNode assertTrue getElseNode getThenNode getElseNode assertEquals getVal assertEquals getVal assertNotNull toString assertNotNull toString assertNotNull toString
test_case.modifiers: @Test public
test_case.parameters: ()
test_case.return: void
test_case.signature: void testFit()
focal_class.file: modules/ml/src/main/java/org/apache/ignite/ml/tree/DecisionTreeClassificationTrainer.java
focal_class.identifier: DecisionTreeClassificationTrainer
focal_class.class_method_signature: DecisionTreeClassificationTrainer.DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease)
focal_class.full_signature: public  DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease)
focal_class.identifier: DecisionTreeClassificationTrainer
focal_class.modifiers: public
focal_class.parameters: (int maxDeep, double minImpurityDecrease)
focal_class.signature:  DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease)
focal_class.class_method_signature: DecisionTreeClassificationTrainer.DecisionTreeClassificationTrainer()
focal_class.full_signature: public  DecisionTreeClassificationTrainer()
focal_class.identifier: DecisionTreeClassificationTrainer
focal_class.modifiers: public
focal_class.parameters: ()
focal_class.signature:  DecisionTreeClassificationTrainer()
focal_class.class_method_signature: DecisionTreeClassificationTrainer.DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease,
          StepFunctionCompressor<GiniImpurityMeasure> compressor)
focal_class.full_signature: public  DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease,
          StepFunctionCompressor<GiniImpurityMeasure> compressor)
focal_class.identifier: DecisionTreeClassificationTrainer
focal_class.modifiers: public
focal_class.parameters: (int maxDeep, double minImpurityDecrease,
          StepFunctionCompressor<GiniImpurityMeasure> compressor)
focal_class.signature:  DecisionTreeClassificationTrainer(int maxDeep, double minImpurityDecrease,
          StepFunctionCompressor<GiniImpurityMeasure> compressor)
focal_class.class_method_signature: DecisionTreeClassificationTrainer.withMaxDeep(Double maxDeep)
focal_class.full_signature: public DecisionTreeClassificationTrainer withMaxDeep(Double maxDeep)
focal_class.identifier: withMaxDeep
focal_class.modifiers: public
focal_class.parameters: (Double maxDeep)
focal_class.return: DecisionTreeClassificationTrainer
focal_class.signature: DecisionTreeClassificationTrainer withMaxDeep(Double maxDeep)
focal_class.class_method_signature: DecisionTreeClassificationTrainer.withMinImpurityDecrease(Double minImpurityDecrease)
focal_class.full_signature: public DecisionTreeClassificationTrainer withMinImpurityDecrease(Double minImpurityDecrease)
focal_class.identifier: withMinImpurityDecrease
focal_class.modifiers: public
focal_class.parameters: (Double minImpurityDecrease)
focal_class.return: DecisionTreeClassificationTrainer
focal_class.signature: DecisionTreeClassificationTrainer withMinImpurityDecrease(Double minImpurityDecrease)
focal_class.class_method_signature: DecisionTreeClassificationTrainer.withUseIndex(boolean useIdx)
focal_class.full_signature: public DecisionTreeClassificationTrainer withUseIndex(boolean useIdx)
focal_class.identifier: withUseIndex
focal_class.modifiers: public
focal_class.parameters: (boolean useIdx)
focal_class.return: DecisionTreeClassificationTrainer
focal_class.signature: DecisionTreeClassificationTrainer withUseIndex(boolean useIdx)
focal_class.class_method_signature: DecisionTreeClassificationTrainer.getMaxDeep()
focal_class.full_signature: public double getMaxDeep()
focal_class.identifier: getMaxDeep
focal_class.modifiers: public
focal_class.parameters: ()
focal_class.return: double
focal_class.signature: double getMaxDeep()
focal_class.class_method_signature: DecisionTreeClassificationTrainer.getMinImpurityDecrease()
focal_class.full_signature: public double getMinImpurityDecrease()
focal_class.identifier: getMinImpurityDecrease
focal_class.modifiers: public
focal_class.parameters: ()
focal_class.return: double
focal_class.signature: double getMinImpurityDecrease()
focal_class.class_method_signature: DecisionTreeClassificationTrainer.getImpurityMeasureCalculator(
          Dataset<EmptyContext, DecisionTreeData> dataset)
focal_class.full_signature: @Override protected ImpurityMeasureCalculator<GiniImpurityMeasure> getImpurityMeasureCalculator(
          Dataset<EmptyContext, DecisionTreeData> dataset)
focal_class.identifier: getImpurityMeasureCalculator
focal_class.modifiers: @Override protected
focal_class.parameters: (
          Dataset<EmptyContext, DecisionTreeData> dataset)
focal_class.return: ImpurityMeasureCalculator<GiniImpurityMeasure>
focal_class.signature: ImpurityMeasureCalculator<GiniImpurityMeasure> getImpurityMeasureCalculator(
          Dataset<EmptyContext, DecisionTreeData> dataset)
focal_class.class_method_signature: DecisionTreeClassificationTrainer.withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
focal_class.full_signature: @Override public DecisionTreeClassificationTrainer withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
focal_class.identifier: withEnvironmentBuilder
focal_class.modifiers: @Override public
focal_class.parameters: (LearningEnvironmentBuilder envBuilder)
focal_class.return: DecisionTreeClassificationTrainer
focal_class.signature: DecisionTreeClassificationTrainer withEnvironmentBuilder(LearningEnvironmentBuilder envBuilder)
focal_class.superclass: extends DecisionTree<GiniImpurityMeasure>
focal_method.body: public DecisionTreeClassificationTrainer withUseIndex(boolean useIdx) {
          this.usingIdx = useIdx;
          return this;
      }
focal_method.class_method_signature: DecisionTreeClassificationTrainer.withUseIndex(boolean useIdx)
focal_method.full_signature: public DecisionTreeClassificationTrainer withUseIndex(boolean useIdx)
focal_method.identifier: withUseIndex
focal_method.modifiers: public
focal_method.parameters: (boolean useIdx)
focal_method.return: DecisionTreeClassificationTrainer
focal_method.signature: DecisionTreeClassificationTrainer withUseIndex(boolean useIdx)