新闻推荐:YoutubeDNN() got an unexpected keyword argument ‘num_sampled‘

 项目来源:零基础入门推荐系统 - 新闻推荐_学习赛_天池大赛-阿里云天池 (aliyun.com) 

报错代码: 

model = YoutubeDNN(user_feature_columns, item_feature_columns,user_dnn_hidden_units=(64, 16),num_sampled=5 )

报错:

 

 修改:


from deepmatch.utils import sampledsoftmaxloss, NegativeSampler
from tensorflow.keras.utils import plot_model

from collections import Counter
train_counter = Counter(train_model_input['click_article_id'])
item_count = [train_counter.get(i,0) for i in range(item_feature_columns[0].vocabulary_size)]
sampler_config = NegativeSampler('frequency',num_sampled=5,item_name="click_article_id",item_count=item_count)

import tensorflow as tf
if tf.__version__ >= '2.0.0':
    tf.compat.v1.disable_eager_execution()
else:
    K.set_learning_phase(True)
    
model = YoutubeDNN(user_feature_columns, item_feature_columns, user_dnn_hidden_units=(64,16, embedding_dim), sampler_config=sampler_config)

原因:

 原项目中用的代码是符合deepmatch==0.1.0的,但是根据包的不断更新,默认安装的deepmatch==0.9.3 比较新,对应的源码中的YoutubeDNN有了一些参数变化,跟着最新的源码修改即可。

  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值