Skip to content

iesl/narrative-driven-rec-mint

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 

Repository files navigation

This code is intended to aid anonymous reviews, it isnt setup for allowin reproducibility efforts yet.

Some files which may be of interest:

src/pre_process/yelppoi_prompts.py: Prompts used for generating synthetic queries and making Grounded LLM predictions.

src/pre_process/pre_proc_yelppoi.py: Pre-processing the user-item interactions and generating synthetic narrative queries.

src/learning/main_fsim.py: Starting point for examining bi-encoder implementations and training.

src/learning/main_ce.py: Starting point for examining cross-encoder implementations and training.

config/models_config/yelppoi: Config files with hyperparameters for bi-encoders and cross-encoders. recinpars refers to the bi-encoder and recinparsce refers to the cross-encoder.

About

Mint: A data augmentation method for narrative driven recommendation.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages