test_class.declarator: useIdx
test_class.modifier: @Parameterized.Parameter
      public
test_class.original_string: @Parameterized.Parameter
      public boolean useIdx;
test_class.type: boolean
test_class.var_name: useIdx
test_class.file: modules/ml/src/test/java/org/apache/ignite/ml/tree/impurity/gini/GiniImpurityMeasureCalculatorTest.java
test_class.identifier: GiniImpurityMeasureCalculatorTest
test_case.body: @Test
      public void testGetLabelCode() {
          Map<Double, Integer> encoder = new HashMap<>();
          encoder.put(0.0, 0);
          encoder.put(1.0, 1);
          encoder.put(2.0, 2);
  
          GiniImpurityMeasureCalculator calculator = new GiniImpurityMeasureCalculator(encoder, useIdx);
  
          assertEquals(0, calculator.getLabelCode(0.0));
          assertEquals(1, calculator.getLabelCode(1.0));
          assertEquals(2, calculator.getLabelCode(2.0));
      }
test_case.class_method_signature: GiniImpurityMeasureCalculatorTest.testGetLabelCode()
test_case.full_signature: @Test public void testGetLabelCode()
test_case.identifier: testGetLabelCode
test_case.invocations: put put put assertEquals getLabelCode assertEquals getLabelCode assertEquals getLabelCode
test_case.modifiers: @Test public
test_case.parameters: ()
test_case.return: void
test_case.signature: void testGetLabelCode()
focal_class.declarator: serialVersionUID = -522995134128519679L
focal_class.modifier: private static final
focal_class.original_string: private static final long serialVersionUID = -522995134128519679L;
focal_class.type: long
focal_class.var_name: serialVersionUID
focal_class.declarator: lbEncoder
focal_class.modifier: private final
focal_class.original_string: private final Map<Double, Integer> lbEncoder;
focal_class.type: Map<Double, Integer>
focal_class.var_name: lbEncoder
focal_class.file: modules/ml/src/main/java/org/apache/ignite/ml/tree/impurity/gini/GiniImpurityMeasureCalculator.java
focal_class.identifier: GiniImpurityMeasureCalculator
focal_class.class_method_signature: GiniImpurityMeasureCalculator.GiniImpurityMeasureCalculator(Map<Double, Integer> lbEncoder, boolean useIdx)
focal_class.full_signature: public  GiniImpurityMeasureCalculator(Map<Double, Integer> lbEncoder, boolean useIdx)
focal_class.identifier: GiniImpurityMeasureCalculator
focal_class.modifiers: public
focal_class.parameters: (Map<Double, Integer> lbEncoder, boolean useIdx)
focal_class.signature:  GiniImpurityMeasureCalculator(Map<Double, Integer> lbEncoder, boolean useIdx)
focal_class.class_method_signature: GiniImpurityMeasureCalculator.calculate(DecisionTreeData data, TreeFilter filter, int depth)
focal_class.full_signature: @SuppressWarnings("unchecked") @Override public StepFunction<GiniImpurityMeasure>[] calculate(DecisionTreeData data, TreeFilter filter, int depth)
focal_class.identifier: calculate
focal_class.modifiers: @SuppressWarnings("unchecked") @Override public
focal_class.parameters: (DecisionTreeData data, TreeFilter filter, int depth)
focal_class.return: StepFunction<GiniImpurityMeasure>[]
focal_class.signature: StepFunction<GiniImpurityMeasure>[] calculate(DecisionTreeData data, TreeFilter filter, int depth)
focal_class.class_method_signature: GiniImpurityMeasureCalculator.getLabelCode(double lb)
focal_class.full_signature:  int getLabelCode(double lb)
focal_class.identifier: getLabelCode
focal_class.parameters: (double lb)
focal_class.return: int
focal_class.signature: int getLabelCode(double lb)
focal_class.superclass: extends ImpurityMeasureCalculator<GiniImpurityMeasure>
focal_method.body: int getLabelCode(double lb) {
          Integer code = lbEncoder.get(lb);
  
          assert code != null : "Can't find code for label " + lb;
  
          return code;
      }
focal_method.class_method_signature: GiniImpurityMeasureCalculator.getLabelCode(double lb)
focal_method.full_signature:  int getLabelCode(double lb)
focal_method.identifier: getLabelCode
focal_method.invocations: get
focal_method.parameters: (double lb)
focal_method.return: int
focal_method.signature: int getLabelCode(double lb)