iesl/narrative-driven-rec-mint

Mint: A data augmentation method for narrative driven recommendation.

8

stars

5

commits

Python

primary language

May 10, 2023

updated

README

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.

Contributors

MSheshera

5 commits

iesl/narrative-driven-rec-mint

Mint: A data augmentation method for narrative driven recommendation.

8

stars

5

commits

Python

primary language

May 10, 2023

updated

README

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.

Contributors

MSheshera

5 commits

Languages

Python

100.0%