Skip to content

Generative Recommendation: Towards Next-generation Recommender Paradigm

Notifications You must be signed in to change notification settings

Linxyhaha/GeneRec

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

GeneRec

This is the pytorch implementation of our paper:

Generative Recommendation: Towards Next-generation Recommender Paradigm

Environment

  • Anaconda 3
  • python 3.8.15
  • pytorch 1.7.0
  • numpy 1.23.5

Usage

AI Editor

The codes for AI Editor are in './code/AI_Editor/' folder, including three tasks 1) thumbnail selection & thumbnail generation, 2) micro-video clipping, and 3) micro-video content editing.

Thumbnail selection

The personalized selected thumbnails can be obtained by running

cd code/AI_Editor
python select_thumbnail.py

Thumbnail generation

The personalized generated thumbnails can be obtained by running

cd code/AI_Editor/Thumbnail_gen
python scripts/generate_thumbnail.py

Micro-video clipping

The personalized clipped micro-videos can be obtained by running

cd code/AI_Editor
python clip_microVid.py

Micro-video content editing

This task requires training, which can be achieved by running

cd code/AI_Editor/Content_editing/scripts
sh train.sh

At inference stage, the personalized edited micro-videos can be obtained by running

cd code/AI_Editor/Content_editing/scripts
sh edit.sh

AI Creator

The codes for AI Creator are in './code/AI_Creator/' folder, including the task of micro-video content creation.

Micro-video content creation

This task also requires training, which can be achieved by running

cd code/AI_Creator/Content_creation/scripts
sh train.sh

At inference stage, the personalized created micro-videos can be obtained by running

cd code/AI_Creator/Content_creation/scripts
sh gen.sh

Data

Please feel free to send a request email for the data. The use of the data should be for research purpose only.

About

Generative Recommendation: Towards Next-generation Recommender Paradigm

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages