本文整理汇总了Python中data.DataProvider.prod2idx方法的典型用法代码示例。如果您正苦于以下问题:Python DataProvider.prod2idx方法的具体用法?Python DataProvider.prod2idx怎么用?Python DataProvider.prod2idx使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类data.DataProvider
的用法示例。
在下文中一共展示了DataProvider.prod2idx方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: str
# 需要导入模块: from data import DataProvider [as 别名]
# 或者: from data.DataProvider import prod2idx [as 别名]
batch = "".join([batch, str(word_embed[idx][i])])
batch = "".join([batch, " "])
batch = "".join([batch, "\n"])
idx += 1
if len(batch) > 100000:
f.write(batch)
batch = ""
f.write(batch)
print("finish generate")
# describe model
# dp.prod2idx issue
# repopulate prod2idx
dp = DataProvider(conf)
dp.prod2idx = {}
for idx, prod in enumerate(dp.idx2prod):
dp.prod2idx[prod] = idx
describe_model = np.load(conf.path_model_npy + ".npy")
word_embed = describe_model[0]
prod_embed = describe_model[1]
transfer_w = describe_model[2]
transfer_b = describe_model[3]
# test doc
model = Word2Vec.load_word2vec_format(conf.path_doc_w2c)
print("init")
while True:
source = input()
source_id = dp.prod2idx[source]