Belle II Software  release-08-01-10
State Class Reference

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 20 of file keras.py.

Constructor & Destructor Documentation

◆ __init__()

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

Definition at line 25 of file keras.py.

25  def __init__(self, model=None, **kwargs):
26  """ Constructor of the state object """
27 
28  self.model = model
29 
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: