(self, sess=None)
| 100 | print("Model saved in file: %s" % save_path) |
| 101 | |
| 102 | def load(self, sess=None): |
| 103 | if not sess: |
| 104 | raise AttributeError("TensorFlow session not provided.") |
| 105 | saver = tf.train.Saver(self.vars) |
| 106 | save_path = "Data/checkpoint/%s.ckpt" % self.name |
| 107 | #save_path = "checks/tmp/%s.ckpt" % self.name |
| 108 | saver.restore(sess, save_path) |
| 109 | print("Model restored from file: %s" % save_path) |
| 110 | |
| 111 | class GCN(Model): |
| 112 | def __init__(self, placeholders, **kwargs): |
no outgoing calls
no test coverage detected