ATLAS Offline Software
Public Member Functions | List of all members
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 23 of file convertXGBoostToRootTree.py.

Member Function Documentation

◆ get_default_left()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_default_left (   self)

Definition at line 58 of file convertXGBoostToRootTree.py.

58  def get_default_left(self):
59  if 'children' not in self:
60  return True
61  elif self.get_nodeid('yes') == self.get_nodeid('missing'):
62  return True
63  else:
64  return False
65 
66 

◆ get_left()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_left (   self)

Definition at line 41 of file convertXGBoostToRootTree.py.

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

◆ get_nodeid()

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

Definition at line 53 of file convertXGBoostToRootTree.py.

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

◆ get_right()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_right (   self)

Definition at line 47 of file convertXGBoostToRootTree.py.

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

◆ get_split_feature()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_split_feature (   self)

Definition at line 29 of file convertXGBoostToRootTree.py.

29  def get_split_feature(self):
30  if 'split' in self:
31  return self['split']
32  else:
33  return -1
34 

◆ get_value()

def util.convertXGBoostToRootTree.XBGoostTextNode.get_value (   self)

Definition at line 35 of file convertXGBoostToRootTree.py.

35  def get_value(self):
36  if 'split_condition' in self:
37  return self['split_condition']
38  else:
39  return self['leaf']
40 

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