Belle II Software development
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

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

Member Data Documentation

◆ collection_keys

collection_keys

list of keys to save

Definition at line 31 of file keras.py.

◆ model

model

keras model

Definition at line 28 of file keras.py.


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