Thanks to visit codestin.com
Credit goes to github.com

Skip to content

lrf23/RecommendGNN

Repository files navigation

机器学习大作业

模型与框架介绍

框架融合

  • 为实现HCCF融合进Rechorus,在 src/models/general 中增加了 HCCF 模型文件。在 src/helpers中增加了 HCCFRunner 文件。
  • 模型优化:HCCFRunnerv2 是使用了动态学习率策略的训练代码。HCCFv2 是调参优化但无损失函数优化的模型版本,HCCFv3是加上InfoNCE损失的模型版本。HCCFv4是针对MovieLens数据集的优化模型版本。
  • 一些测评结果在 log
  • 模型权值保存在 model

声明

本仓库融合了Rechorus框架和HCCF模型,仅供学习使用,部分使用到的源代码地址如下所示:

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published