資源簡介
參考文獻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
- 上一篇:樁位坐標提取插件
- 下一篇:2018軟件工程最新畢設題目
評論
共有 條評論