lleaves.compiler.ast.nodes module

class lleaves.compiler.ast.nodes.DecisionNode(idx: int, split_feature: int, threshold: int, decision_type: lleaves.compiler.utils.DecisionType, left_idx: int, right_idx: int)

Bases: Node

add_children(left, right)

cat_threshold: List[int] = None

decision_type: DecisionType

finalize_categorical(cat_threshold)

idx: int

left: Node = None

left_idx: int

right: Node = None

right_idx: int

split_feature: int

threshold: int

validate()

class lleaves.compiler.ast.nodes.Forest(trees: List[lleaves.compiler.ast.nodes.Tree], features: list, n_classes: int, objective_func: str, objective_func_config: str, raw_score: bool = False, average_output: bool = False)

Bases: object

average_output: bool = False

features: list

property n_args

n_classes: int

objective_func: str

objective_func_config: str

raw_score: bool = False

trees: List[Tree]

class lleaves.compiler.ast.nodes.LeafNode(idx: int, value: float)

Bases: Node

idx: int

value: float

class lleaves.compiler.ast.nodes.Node

Bases: object

property is_leaf

class lleaves.compiler.ast.nodes.Tree(idx: int, root_node: lleaves.compiler.ast.nodes.Node, features: list, class_id: int)

Bases: object

class_id: int

features: list

idx: int

root_node: Node