YoutubeNet推荐网络的Tensorflow实现

参考文献Deep Neural Networks for Youtube Recommendation的tensorflow实现代码。数据集：MovieLens-20M，构造方式，以user id的观看记录时间戳构造每个user的观看历史，即每一行为n个video的id以及其所对应的时间戳。video id的label映射是按照其出现的频率做反比

import os
os.environ[‘CUDA_VISIBLE_DEVICES‘] = ‘7‘
import time
from data.get_ml import ucf_data  
import tensorflow as tf
import numpy as np
from model_ml import video_model
from data import config as cfg
slim = tf.contrib.slim


class Multi_Trainer（object）:

    def __init__（selfmodeldata1）:

        self.batch_size = cfg.train_batch_size
        self.model = model
        self.data1 = data1
        self.num_gpus = 1
        self.num_classes = data1.num_classes
        
    def tower_loss（self history ex_age labels）:
        net logit losses = self.model.youtube_network（history ex_age  self.num_classeslabels）
        regularization_losses = tf.reduce_mean（tf.get_collection（tf.GraphKeys.REGULARIZATION_LOSSES））
        total_loss = losses + regularization_l

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----

    .......       282  2019-04-11 20:28  data\config.py

    .......      3808  2019-04-11 20:28  data\get_ml.py

    .......         0  2019-04-11 20:28  data\__init__.py

    .......        96  2019-04-11 20:28  .gitignore

    .......         0  2019-04-11 20:28  __init__.py

     文件       7380  2019-04-11 20:36  ml_train.py

     文件       2462  2019-04-11 20:36  model_ml.py

     目录          0  2019-04-11 20:36  data

----------- ---------  ---------- -----  ----

                14028                    8