Belle II Software development
State Class Reference

Public Member Functions

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

Public Attributes

 model = model
 keras model
 
list collection_keys = []
 list of keys to save
 

Detailed Description

keras state

Definition at line 23 of file keras.py.

Constructor & Destructor Documentation

◆ __init__()

__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

list collection_keys = []

list of keys to save

Definition at line 34 of file keras.py.

◆ model

model = model

keras model

Definition at line 31 of file keras.py.


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