Skip to content

koc-lab/grte

Repository files navigation

Graph Receptive Transformer Encoder for Text Classification

GitHub License Python Version PyTorch

This repository contains the code for the "Graph Receptive Transformer Encoder for Text Classification" paper published in IEEE Transactions on Signal and Information Processing over Networks. Please cite the following paper if you use this code in your research:

@article{grte2024,
  author   = {Aras, Arda Can and Alikaşifoğlu, Tuna and Koç, Aykut},
  journal  = {IEEE Transactions on Signal and Information Processing over Networks},
  title    = {Graph Receptive Transformer Encoder for Text Classification},
  year     = {2024},
  volume   = {10},
  number   = {},
  pages    = {347-359},
  doi      = {10.1109/TSIPN.2024.3380362}
}

Installation

This codebase utilizes Poetry for package management. To install the dependencies, in the root directory run:

poetry install

or if you do not want to use Poetry, or use the exact pinned versions, you can install the dependencies provided in requirements.txt using pip or conda, e,g.,

pip install -r requirements.txt