Belle II Software development
State Class Reference
Inheritance diagram for State:

Public Member Functions

def __init__ (self, bst=None, params=None, X_valid=None, y_valid=None, path='LGBM.txt', trainFraction=0.8, num_round=100)
 

Public Attributes

 X_valid
 X valid.
 
 y_valid
 Y valid.
 
 path
 weights save path
 
 params
 LightGBM Model parameter.
 
 bst
 saved Best model
 
 trainFraction
 train fraction for dataset splitting
 

Detailed Description

LGBM state

Definition at line 8 of file lightgbm.py.

Constructor & Destructor Documentation

◆ __init__()

def __init__ (   self,
  bst = None,
  params = None,
  X_valid = None,
  y_valid = None,
  path = 'LGBM.txt',
  trainFraction = 0.8,
  num_round = 100 
)
 Constructor of the state object 

Definition at line 13 of file lightgbm.py.

13 def __init__(self, bst=None, params=None, X_valid=None, y_valid=None, path='LGBM.txt', trainFraction=0.8, num_round=100):
14 """ Constructor of the state object """
15
16 self.X_valid = X_valid
17
18 self.y_valid = y_valid
19
20 self.path = path
21
22 self.params = params
23
24 self.bst = bst
25
26 self.trainFraction = trainFraction
27
28

Member Data Documentation

◆ bst

bst

saved Best model

Definition at line 24 of file lightgbm.py.

◆ params

params

LightGBM Model parameter.

Definition at line 22 of file lightgbm.py.

◆ path

path

weights save path

Definition at line 20 of file lightgbm.py.

◆ trainFraction

trainFraction

train fraction for dataset splitting

Definition at line 26 of file lightgbm.py.

◆ X_valid

X_valid

X valid.

Definition at line 16 of file lightgbm.py.

◆ y_valid

y_valid

Y valid.

Definition at line 18 of file lightgbm.py.


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