Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

inference.py报错 #1

Open
wjxfoxx opened this issue Mar 2, 2024 · 8 comments
Open

inference.py报错 #1

wjxfoxx opened this issue Mar 2, 2024 · 8 comments

Comments

@wjxfoxx
Copy link

wjxfoxx commented Mar 2, 2024

Hello author, I am currently reproducing your TDRO model code, but I found an unnecessary parameter num_period in the train_dataset function call at line 117 in the original inference.py, whether I deleted the parameter or added it in Dataset.py An error will be reported, or the --no_cuda option is not defined, or there is an error in the eval() function, etc. How should I solve it?

@coralineyang
Copy link

请问你成功复现了吗?我发现这个Inference代码好像有问题,model.result好像也没说是哪里来的。

@wjxfoxx
Copy link
Author

wjxfoxx commented Mar 12, 2024 via email

@Linxyhaha
Copy link
Owner

Hello author, I am currently reproducing your TDRO model code, but I found an unnecessary parameter num_period in the train_dataset function call at line 117 in the original inference.py, whether I deleted the parameter or added it in Dataset.py An error will be reported, or the --no_cuda option is not defined, or there is an error in the eval() function, etc. How should I solve it?

你好,感谢对我们工作的关注! 我们更正了一下inference.py 代码中的typo,现在应该可以跑通了 :)

@Linxyhaha
Copy link
Owner

请问你成功复现了吗?我发现这个Inference代码好像有问题,model.result好像也没说是哪里来的。

你好,我们更正了inference.py代码中的typo,现在应该没有问题了。另外model.result是在model_CLCRec.py中定义的,并在每个epoch后会更新一次 (Train.py line81-83). :)

@MrShouxingMa
Copy link

作者您好,请问如果我要更换数据集,如何在最基本的数据集基础上生成代码所需数据?

@Linxyhaha
Copy link
Owner

作者您好,请问如果我要更换数据集,如何在最基本的数据集基础上生成代码所需数据?

你好,我们新提供了一个数据集的说明文件在data/ 下,包含了数据集文件的内容以及格式。如果需要更换数据集,确保包含interactions, item timestamps, item feature, 以及warm/cold sets 这些数据文件(在数据集说明文件中有详细介绍)。

@MrShouxingMa
Copy link

感谢作者的及时回复,我还想请请教,如果数据集中没有项目时间戳,
只包含最基本的用户和项目交互数据,暖/冷集合以及项目的特征,我想知道是否可以运行此模型?

@Linxyhaha
Copy link
Owner

感谢作者的及时回复,我还想请请教,如果数据集中没有项目时间戳, 只包含最基本的用户和项目交互数据,暖/冷集合以及项目的特征,我想知道是否可以运行此模型?

如果没有时间戳的话,可以考虑使用我们代码中对kwai数据集的处理方式:利用用户交互产品的相对时间顺序。具体可以参考Data.py这个脚本中的 gen_env_relative(self, train_data, N) 函数。需要注意的是这种处理方式仍然需要知道用户序列中产品的相对时间顺序。

我们的方法需要将用户的训练数据划分成不同的period来利用好shifting factor,因此需要有绝对时间戳或相对交互顺序。具体可以参考论文中的Section 4.1 以及公式5。如果没有时间信息,也可以使用我们的模型,但此时period=1,shifting factor将会变成常数。

@Linxyhaha Linxyhaha mentioned this issue Jun 30, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants