Pinned Loading

1. Conditional-Variational-Autoencoder-Recommendation Conditional-Variational-Autoencoder-Recommendation Public

   PyTorch implementation of our paper "Improving Item Cold-start Recommendation via Model-agnostic Conditional Variational Autoencoder" accepted by SIGIR 2022.

   Python 4 1

2. DML DML Public

   本文提出了一种基于注意力机制的多任务关联关系学习模型(DML)。从原始输入特征信息中学习到task的表达，借用Attention机制实现task与experts间的关联，实现对特征的聚焦，对不同任务之间进一步通过co-Attention的方式实现上层表达的共享，捕捉任务间的关联。模型如图所示：改进后论文已中论文Deep Mutual Learning across Task Towers f…

   Python 2

3. Slate-Aware-Ranking Slate-Aware-Ranking Public

   The implementation for our paper "Slate-Aware Ranking for Recommendation" accepted by WSDM.23

   Python

4. Visualglm-image-to-text Visualglm-image-to-text Public

   补充了一些Visualglm缺少的文件，可以对Visualglm进行训练，实例中是对人脸做了面相的识别

   Python 12

5. ChatGLM-RLHF-LoRA-RM-PPO ChatGLM-RLHF-LoRA-RM-PPO Public

   ChatGLM-6B添加了RLHF的实现，以及部分核心代码的逐行讲解 ,实例部分是做了个新闻短标题的生成，以及指定context推荐的RLHF的实现

   Python 74 7

6. vqvae-dalle vqvae-dalle Public

   使用vqvae 和简易 dalle 结构 进行用户和物品冷启动

   Python