Exemplo n.º 1
0
def _get_root_node(tree: DecisionTreeClassificationModel):
    return tree._call_java('rootNode')
Exemplo n.º 2
0
def _get_root_node(tree: DecisionTreeClassificationModel):
    if hasattr(tree, 'trees'):
        return tree.trees[0]._call_java('rootNode')
    else:
        return tree._call_java('rootNode')