From a1957c1148be1a036825687c211881c9bbe07ed3 Mon Sep 17 00:00:00 2001
From: xjqbest <173596896@qq.com>
Date: Fri, 29 May 2020 10:57:33 +0800
Subject: [PATCH] fix

---
 core/model.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/core/model.py b/core/model.py
index 5aae5c52..9422dc49 100755
--- a/core/model.py
+++ b/core/model.py
@@ -159,6 +159,8 @@ class Model(object):
         name = "dataset." + kwargs.get("dataset_name") + "."
         sparse_slots = envs.get_global_env(name + "sparse_slots", "").strip()
         dense_slots = envs.get_global_env(name + "dense_slots", "").strip()
+        self._sparse_data_var_map = {}
+        self._dense_data_var_map = {}
         if sparse_slots != "" or dense_slots != "":
             if sparse_slots == "":
                 sparse_slots = []
@@ -181,12 +183,14 @@ class Model(object):
                     dtype="float32")
                 data_var_.append(l)
                 self._dense_data_var.append(l)
+                self._dense_data_var_map[dense_slots[i]] = l
             self._sparse_data_var = []
             for name in sparse_slots:
                 l = fluid.layers.data(
                     name=name, shape=[1], lod_level=1, dtype="int64")
                 data_var_.append(l)
                 self._sparse_data_var.append(l)
+                self._sparse_data_var_map[name] = l
             return data_var_
 
         else:
-- 
GitLab