Skip to content

DCY1117/LuckyLook

 
 

Repository files navigation

LuckyLook

LuckyLook is a tool designed for recommending PubMed scientific journals for researchers looking to publish their work. It uses a Transformer model with additional graph-level information to provide a better recommendation.

Model training

Dataset Setup

  1. Download the dataset from Zenodo: https://doi.org/10.5281/zenodo.8386011
  2. Create a Dataset/ folder in the LuckyLook directory.
  3. Move the downloaded dataset into the Dataset/ folder.

Usage

Configuration JSON files, which include parameters and settings for model training, are stored in the configs/ directory.

Training

To train the model, run:

python train.py --config configs/<config_file>.json

Testing

For testing, specify the checkpoint directory:

python test.py --resume path/to/checkpoint

Streamlit web application

Usage

streamlit run LuckyLook.py

Citation

Acknowledgements

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 68.1%
  • Python 31.9%