From 0cc5872dfe192e0cc541d60076b695460d905b05 Mon Sep 17 00:00:00 2001
From: chengmo <chengmo@baidu.com>
Date: Wed, 6 May 2020 16:55:10 +0800
Subject: [PATCH] fix reader

---
 models/recall/tdm/tdm_reader.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/recall/tdm/tdm_reader.py b/models/recall/tdm/tdm_reader.py
index 32d33aeb..17413249 100644
--- a/models/recall/tdm/tdm_reader.py
+++ b/models/recall/tdm/tdm_reader.py
@@ -33,8 +33,8 @@ class TrainReader(Reader):
             This function needs to be implemented by the user, based on data format
             """
             features = (line.strip('\n')).split('\t')
-            input_emb = features[0].split(' ')
-            item_label = [features[1]]
+            input_emb = map(float, features[0].split(' '))
+            item_label = [int(features[1])]
 
             feature_name = ["input_emb", "item_label"]
             yield zip(feature_name, [input_emb] + [item_label])
-- 
GitLab