-
大小: 5KB文件類型: .rar金幣: 1下載: 0 次發(fā)布日期: 2021-01-09
- 語言: 其他
- 標(biāo)簽: tensor??youtub??推薦網(wǎng)絡(luò)??
資源簡(jiǎn)介
參考文獻(xiàn)Deep Neural Networks for Youtube Recommendation的tensorflow實(shí)現(xiàn)代碼。數(shù)據(jù)集:MovieLens-20M,構(gòu)造方式,以u(píng)ser id的觀看記錄時(shí)間戳構(gòu)造每個(gè)user的觀看歷史,即每一行為n個(gè)video的id以及其所對(duì)應(yīng)的時(shí)間戳。video id的label映射是按照其出現(xiàn)的頻率做反比
代碼片段和文件信息
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
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
????.......???????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
- 上一篇:樁位坐標(biāo)提取插件
- 下一篇:2018軟件工程最新畢設(shè)題目
評(píng)論
共有 條評(píng)論