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

Constructor & Destructor Documentation

◆ __init__()

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

Definition at line 28 of file keras.py.

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

Member Data Documentation

◆ collection_keys

collection_keys

list of keys to save

Definition at line 34 of file keras.py.

◆ model

model

keras model

Definition at line 31 of file keras.py.


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