資源簡介
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
評論
共有 條評論