YoutubeNet深度学习召回模型tf实现.zip

YoutubeNet深度学习召回模型tf实现 原文：Deep Neural Networks for Youtube Recommendation

import random
import pickle
import numpy as np
import time
random.seed（1234）


start_time = time.time（）

with open（‘remap.pkl‘ ‘rb‘） as f:
    userclick_data = pickle.load（f）
    item_key brand_key msort_key user_key = pickle.load（f）
    brand_list = pickle.load（f）
    msort_list = pickle.load（f）
    user_count item_count brand_countmsort_count example_count = pickle.load（f）

    print（‘user_count: %d\titem_count: %d\tbrand_count: %d\texample_count: %d‘ %
      （user_count item_count brand_count example_count））
    train_set = []
    test_set = []
    uid_num=0
    for UId hist in userclick_data.groupby（‘UId‘）:
        uid_num+=1

        pos_list = hist[‘ItemId‘].tolist（）

        if len（pos_list）<3:
            continue
  
        def gen_neg（）:
            neg = pos_list[0]

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-09-04 02:41  YoutubeNet-master\
     文件        1203  2018-09-04 02:41  YoutubeNet-master\.gitignore
     文件          57  2018-09-04 02:41  YoutubeNet-master\README.md
     目录           0  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\
     目录           0  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\__pycache__\
     文件        3904  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\__pycache__\input.cpython-35.pyc
     文件        4570  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\__pycache__\model.cpython-35.pyc
     文件        2429  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\build_model_data.py
     目录           0  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\data\
     文件        1825  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\data\creat_session_data.py
     文件         877  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\data\get_click_data.py
     文件        2242  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\evalu_.py
     文件        3672  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\input.py
     文件        4213  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\main.py
     文件        1847  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\map_id_idx.py
     文件        5893  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\model.py
     文件        8347  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\model_rnn.py
     文件        1652  2018-09-04 02:41  YoutubeNet-master\YoutubeNet_pkl\offline_criteria.py