Skip to content

Commit 0a63c76

Browse files
committedOct 28, 2019
include_zero removed from make_ites)tensor
1 parent 84c47d9 commit 0a63c76

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed
 

‎recnn/data/utils.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -128,9 +128,7 @@ def prepare_batch_static_size(batch, item_embeddings_tensor=False, frame_size=10
128128
# id_to_key - dict index -> key
129129

130130

131-
def make_items_tensor(items_embeddings_key_dict, include_zero=True):
132-
if include_zero:
133-
items_embeddings_key_dict[0] = torch.zeros([128, ])
131+
def make_items_tensor(items_embeddings_key_dict):
134132
keys = list(sorted(items_embeddings_key_dict.keys()))
135133
key_to_id = dict(zip(keys, range(len(keys))))
136134
id_to_key = dict(zip(range(len(keys)), keys))

0 commit comments

Comments
 (0)
Please sign in to comment.