util.convertLGBMToRootTree.LGBMTextNode Class Reference

Inheritance diagram for util.convertLGBMToRootTree.LGBMTextNode:

Collaboration diagram for util.convertLGBMToRootTree.LGBMTextNode:

Public Member Functions
	__init__ (self, structure, invert_as_tmva=False)
	get_split_feature (self)
	get_value (self)
	get_left (self)
	get_right (self)
	get_default_left (self)

Public Attributes
	invert_as_tmva = invert_as_tmva

Protected Member Functions
	_get_left (self)
	_get_right (self)

Detailed Description

Adaptor from LGBM dictionary to tree

Definition at line 48 of file convertLGBMToRootTree.py.

Constructor & Destructor Documentation

__init__()

util.convertLGBMToRootTree.LGBMTextNode.__init__	(		self,
			structure,
			invert_as_tmva = False )

Definition at line 53 of file convertLGBMToRootTree.py.

    def __init__(self, structure, invert_as_tmva=False):
        super(LGBMTextNode, self).__init__(structure)
        self.invert_as_tmva = invert_as_tmva
 

Member Function Documentation

_get_left()

util.convertLGBMToRootTree.LGBMTextNode._get_left ( self )

protected

Definition at line 69 of file convertLGBMToRootTree.py.

    def _get_left(self):
        if "left_child" not in self:
            return None
        if self["decision_type"] == "<=":
            return LGBMTextNode(self["left_child"])
        else:
            return LGBMTextNode(self["right_child"])
 

_get_right()

util.convertLGBMToRootTree.LGBMTextNode._get_right ( self )

protected

Definition at line 77 of file convertLGBMToRootTree.py.

    def _get_right(self):
        if "right_child" not in self:
            return None
        if self["decision_type"] == "<=":
            return LGBMTextNode(self["right_child"])
        else:
            return LGBMTextNode(self["left_child"])
 

get_default_left()

util.convertLGBMToRootTree.LGBMTextNode.get_default_left

(

self

)

Definition at line 97 of file convertLGBMToRootTree.py.

    def get_default_left(self):
        return self.get("default_left", True)
 
 

get_left()

util.convertLGBMToRootTree.LGBMTextNode.get_left

(

self

)

Definition at line 85 of file convertLGBMToRootTree.py.

    def get_left(self):
        if not self.invert_as_tmva:
            return self._get_left()
        else:
            return self._get_right()
 

get_right()

util.convertLGBMToRootTree.LGBMTextNode.get_right

(

self

)

Definition at line 91 of file convertLGBMToRootTree.py.

    def get_right(self):
        if not self.invert_as_tmva:
            return self._get_right()
        else:
            return self._get_left()
 

get_split_feature()

util.convertLGBMToRootTree.LGBMTextNode.get_split_feature

(

self

)

Definition at line 57 of file convertLGBMToRootTree.py.

    def get_split_feature(self):
        if "split_feature" in self:
            return self["split_feature"]
        else:
            return -1
 

get_value()

util.convertLGBMToRootTree.LGBMTextNode.get_value

(

self

)

Definition at line 63 of file convertLGBMToRootTree.py.

    def get_value(self):
        if "threshold" in self:
            return self["threshold"]
        else:
            return self["leaf_value"]
 

Member Data Documentation

invert_as_tmva

util.convertLGBMToRootTree.LGBMTextNode.invert_as_tmva = invert_as_tmva

Definition at line 55 of file convertLGBMToRootTree.py.

---

The documentation for this class was generated from the following file:

• convertLGBMToRootTree.py