Belle II Software  release-06-01-15
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 19 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 24 of file contrib_keras.py.

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

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