Belle II Software  release-05-02-19
State Class Reference
Inheritance diagram for State:
Collaboration diagram for State:

Public Member Functions

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

Public Attributes

 model
 keras model
 
 custom_objects
 used by keras to load custom objects like custom layers
 
 collection_keys
 list of keys to save
 

Detailed Description

Tensorflow state

Definition at line 15 of file contrib_keras.py.

Constructor & Destructor Documentation

◆ __init__()

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

Definition at line 20 of file contrib_keras.py.

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

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