From 6d8ff0ba088741c179a971b987e8a00e87ce6dd7 Mon Sep 17 00:00:00 2001 From: malin10 Date: Tue, 2 Jun 2020 14:19:28 +0800 Subject: [PATCH] bug fix --- models/recall/word2vec/model.py | 21 --------------------- 1 file changed, 21 deletions(-) diff --git a/models/recall/word2vec/model.py b/models/recall/word2vec/model.py index 5822c90f..43417e60 100755 --- a/models/recall/word2vec/model.py +++ b/models/recall/word2vec/model.py @@ -186,27 +186,6 @@ class Model(ModelBase): staircase=True)) return optimizer - def analogy_input(self): - sparse_feature_number = envs.get_global_env( - "hyper_parameters.sparse_feature_number", None, self._namespace) - self.analogy_a = fluid.data( - name="analogy_a", shape=[None], dtype='int64') - self.analogy_b = fluid.data( - name="analogy_b", shape=[None], dtype='int64') - self.analogy_c = fluid.data( - name="analogy_c", shape=[None], dtype='int64') - self.analogy_d = fluid.data( - name="analogy_d", shape=[None], dtype='int64') - self._infer_data_var = [ - self.analogy_a, self.analogy_b, self.analogy_c, self.analogy_d - ] - - self._infer_data_loader = fluid.io.DataLoader.from_generator( - feed_list=self._infer_data_var, - capacity=64, - use_double_buffer=False, - iterable=False) - def infer_net(self, inputs): def embedding_layer(input, table_name, initializer_instance=None): emb = fluid.embedding( -- GitLab