util.convertXGBoostToRootTree.XBGoostTextNode Class Reference

Inheritance diagram for util.convertXGBoostToRootTree.XBGoostTextNode:

Collaboration diagram for util.convertXGBoostToRootTree.XBGoostTextNode:

Public Member Functions
def	get_split_feature (self)
def	get_value (self)
def	get_left (self)
def	get_right (self)
def	get_nodeid (self, node_type)
def	get_default_left (self)

Detailed Description

Adaptor from XGBoost dictionary to tree
   * XGboost Yes is < and No is >=

Definition at line 22 of file convertXGBoostToRootTree.py.

Member Function Documentation

get_default_left()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_default_left

(

self

)

Definition at line 57 of file convertXGBoostToRootTree.py.

    def get_default_left(self):
        if 'children' not in self:
            return True
        elif self.get_nodeid('yes') == self.get_nodeid('missing'):
            return True
        else:
            return False
 
 

get_left()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_left

(

self

)

Definition at line 40 of file convertXGBoostToRootTree.py.

    def get_left(self):
        if 'children' not in self:
            return None
        else:  # XGBoost "YES" is left branch in MVAUtils
            return XBGoostTextNode(self['children'][self.get_nodeid('yes')])
 

get_nodeid()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_nodeid	(		self,
			node_type
	)

Definition at line 52 of file convertXGBoostToRootTree.py.

    def get_nodeid(self, node_type):
        for idx, children in enumerate(self['children']):
            if children['nodeid'] == self[node_type]:
                return idx
 

get_right()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_right

(

self

)

Definition at line 46 of file convertXGBoostToRootTree.py.

    def get_right(self):
        if 'children' not in self:
            return None
        else:  # XGBoost "NO" is right branch in MVAUtils
            return XBGoostTextNode(self['children'][self.get_nodeid('no')])
 

get_split_feature()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_split_feature

(

self

)

Definition at line 28 of file convertXGBoostToRootTree.py.

    def get_split_feature(self):
        if 'split' in self:
            return self['split']
        else:
            return -1
 

get_value()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_value

(

self

)

Definition at line 34 of file convertXGBoostToRootTree.py.

    def get_value(self):
        if 'split_condition' in self:
            return self['split_condition']
        else:
            return self['leaf']
 

---

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

• convertXGBoostToRootTree.py