public function buildTree(array $criteria = [])
{
    $treeNode = new TreeNode();
    $checkClass = $this->isBelongToOneClass($criteria);

    if ($checkClass['return']) {
        $treeNode->setAttribute($this->targetAttribute);
        $treeNode->addChild('result', $checkClass['class']);
        $treeNode->setIsLeaf(true);

        return $treeNode;
    }

    $splitCriterion = $this->calculateSplitCriterion($criteria);

    $bestAttrName = $this->getBiggestArrayAttribute($splitCriterion);
    $bestAttrValues = $this->getAttributeValues($bestAttrName);

    $treeNode->setAttribute($bestAttrName);

    unset($splitCriterion[$bestAttrName]);

    foreach ($bestAttrValues as $value) {
        $criteria[$bestAttrName] = $value;
        $targetCount = $this->countTargetByCriteria($criteria);
        $treeNode->addClassesCount($value, $targetCount);

        if (array_sum($targetCount) == 0) {
            $targetCount2 = $this->countTargetByCriteria([$bestAttrName => $value]);
            $biggestClass = $this->getBiggestArrayAttribute($targetCount2);

            $child = new TreeNode();
            $child->setParent($treeNode);
            $child->setAttribute($this->targetAttribute);
            $child->addChild('result', $biggestClass);
            $child->setIsLeaf(true);

            $treeNode->addChild($value, $child);
        } elseif (!empty($splitCriterion)) {
            $child = $this->buildTree($criteria);
            $child->setParent($treeNode);

            $treeNode->addChild($value, $child);
        } else {
            $classProb = $this->calculateClassProbability($criteria);
            $biggestClass = $this->getBiggestArrayAttribute($classProb);

            $child = new TreeNode();
            $child->setParent($treeNode);
            $child->setAttribute($this->targetAttribute);
            $child->addChild('result', $biggestClass);
            $child->setIsLeaf(true);

            $treeNode->addChild($value, $child);
        }
    }

    return $treeNode;
}