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)