def load_tfparts(batch_size=128, dim=64, this_data=None, save_path='this-transe.ckpt', L1=False): tf_parts = transe.TFParts(num_rels=this_data.num_rels(), num_cons=this_data.num_cons(), dim=dim, batch_size=self.batch_size, L1=L1) with tf.Session() as sess: tf_parts._saver.restore(sess, save_path)
def build(self, save_path='this-model.ckpt', data_save_path='this-data.bin'): self.this_data = data.Data() self.this_data.load(data_save_path) self.tf_parts = transe.TFParts(num_rels=self.this_data.num_rels(), num_cons=self.this_data.num_cons(), dim=self.this_data.dim, batch_size=self.this_data.batch_size, L1=self.this_data.L1) sess = tf.Session() self.tf_parts._saver.restore(sess, save_path) # load it value_ht, value_r = sess.run([self.tf_parts._ht, self.tf_parts._r]) # extract values. sess.close() self.vec_c = np.array(value_ht) self.vec_r = np.array(value_r)
def build(self, data, dim=64, batch_size=128, save_path='this-transe.ckpt', data_save_path='this-data.bin', L1=False): self.this_data = data self.dim = self.this_data.dim = dim self.batch_size = self.this_data.batch_size = batch_size self.data_save_path = data_save_path self.save_path = save_path self.L1 = self.this_data.L1 = L1 self.tf_parts = transe.TFParts(num_rels=self.this_data.num_rels(), num_cons=self.this_data.num_cons(), dim=dim, batch_size=self.batch_size, L1=self.L1)