-
Notifications
You must be signed in to change notification settings - Fork 139
Description
I am looking for something equivalent to model.predict(X_test) in keras.
res = dbm.reconstruct(X_test)
Traceback (most recent call last):
File "/home/yunus/PycharmProjects/boltz_/boltzmann-machines-clone/examples/dbm_mnist.py", line 402, in
main()
File "/home/yunus/PycharmProjects/boltz_/boltzmann-machines-clone/examples/dbm_mnist.py", line 377, in main
res = dbm.reconstruct(X_test)
File "/home/yunus/PycharmProjects/boltz_/boltzmann-machines-clone/boltzmann_machines/base/tf_model.py", line 28, in wrapped_f
res = f(model, *args, **kwargs)
File "/home/yunus/PycharmProjects/boltz_/boltzmann-machines-clone/boltzmann_machines/dbm.py", line 877, in reconstruct
self._reconstruction = tf.get_collection('reconstruction')[0]
IndexError: list index out of range
tf.get_collection('reconstruction') returns an empty list.
when i run
dummy2 = tf.get_default_graph().get_all_collection_keys() there is no reconstruction entry.
reconstruct is the equivalent i thought but i cant get it to work.
Is there a reconstruct for RBM or how is it called?