From 8aaeab15dc1d4f0387084b07fce77d55d6dc36de Mon Sep 17 00:00:00 2001 From: wangguibao Date: Thu, 14 Nov 2019 19:05:40 +0800 Subject: [PATCH] elastic CTR --- elastic-ctr/client/demo/elastic_ctr.py | 26 ++++++++++++++++++++------ 1 file changed, 20 insertions(+), 6 deletions(-) diff --git a/elastic-ctr/client/demo/elastic_ctr.py b/elastic-ctr/client/demo/elastic_ctr.py index 77ae6879..75684052 100644 --- a/elastic-ctr/client/demo/elastic_ctr.py +++ b/elastic-ctr/client/demo/elastic_ctr.py @@ -14,7 +14,6 @@ from __future__ import print_function import json -import httplib import sys import os @@ -27,6 +26,13 @@ CTR_EMBEDDING_TABLE_SIZE = 400000001 SLOTS = [] +def str2long(str): + if sys.version_info[0] == 2: + return long(str) + elif sys.version_info[0] == 3: + return int(str) + + def data_reader(data_file, samples, labels): if not os.path.exists(data_file): print("Path %s not exist" % data_file) @@ -45,10 +51,14 @@ def data_reader(data_file, samples, labels): for i in range(0, len(features)): if slots[i] in sample: - sample[slots[i]] = [sample[slots[i]] + long(features[i])] + sample[slots[i]] = [ + sample[slots[i]] + str2long(features[i]) % + CTR_EMBEDDING_TABLE_SIZE + ] else: - sample[slots[ - i]] = [long(features[i]) % CTR_EMBEDDING_TABLE_SIZE] + sample[slots[i]] = [ + str2long(features[i]) % CTR_EMBEDDING_TABLE_SIZE + ] for x in SLOTS: if not x in sample: @@ -85,8 +95,12 @@ if __name__ == "__main__": for sample in batch: instance = api.add_instance() kv = [] - for k, v in sample.iteritems(): - api.add_slot(instance, k, v) + if sys.version_info[0] == 2: + for k, v in sample.iteritems(): + api.add_slot(instance, k, v) + elif sys.version_info[0] == 3: + for k, v in sample.items(): + api.add_slot(instance, k, v) ret = api.inference() print(ret) -- GitLab