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

About

Software for scientific journal recommendation

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 68.1%
  • Python 31.9%