Belle II Software  light-2205-abys
State Class Reference
Inheritance diagram for State:
Collaboration diagram for State:

Public Member Functions

def __init__ (self, model=None, **kwargs)
 

Public Attributes

 model
 keras model
 
 collection_keys
 list of keys to save
 

Detailed Description

Tensorflow.keras state

Definition at line 19 of file keras.py.

Constructor & Destructor Documentation

◆ __init__()

def __init__ (   self,
  model = None,
**  kwargs 
)
 Constructor of the state object 

Definition at line 24 of file keras.py.

24  def __init__(self, model=None, **kwargs):
25  """ Constructor of the state object """
26 
27  self.model = model
28 
29 
30  self.collection_keys = []
31 
32  # other possible things to save into a tensorflow collection
33  for key, value in kwargs.items():
34  self.collection_keys.append(key)
35  setattr(self, key, value)
36 
37 

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