Tensorflow state
Definition at line 18 of file tensorflow.py.
◆ __init__()
def __init__ |
( |
|
self, |
|
|
|
x = None , |
|
|
|
y = None , |
|
|
|
activation = None , |
|
|
|
cost = None , |
|
|
|
optimizer = None , |
|
|
|
session = None , |
|
|
** |
kwargs |
|
) |
| |
Constructor of the state object
Definition at line 23 of file tensorflow.py.
23 def __init__(self, x=None, y=None, activation=None, cost=None, optimizer=None, session=None, **kwargs):
24 """ Constructor of the state object """
30 self.activation = activation
34 self.optimizer = optimizer
36 self.session = session
38 self.collection_keys = [
'x',
'y',
'activation',
'cost',
'optimizer']
41 for key, value
in kwargs.items():
42 self.collection_keys.append(key)
43 setattr(self, key, value)
◆ add_to_collection()
def add_to_collection |
( |
|
self | ) |
|
Add the stored members to the current tensorflow collection
Definition at line 45 of file tensorflow.py.
◆ get_from_collection()
def get_from_collection |
( |
|
self, |
|
|
|
collection_keys = None |
|
) |
| |
Get members from the current tensorflow collection
Definition at line 58 of file tensorflow.py.
The documentation for this class was generated from the following file: