Skip to content

Latest commit

 

History

History
35 lines (29 loc) · 1.21 KB

README.md

File metadata and controls

35 lines (29 loc) · 1.21 KB
Raw

Deep Attributed Graph Embedding

This the official repository of DAGE (Deep Attributed Graph Embedding), where both the source code and the datasets are available. DAGE has been published at MDAI 2022, click here to read the paper.

If you decide to use this resource, please cite:

::

@inproceedings{fersini2022,
    title={Deep Attributed Graph Embeddings},
    author={Fersini, Elisabetta and Mottadelli, Simone and Carbonera, Michele and Messina, Enza},
    year={2022},
    booktitle={Modeling Decisions for Artificial Intelligence},
    year = "2022",
    publisher = "Springer International Publishing",
    pages = "181--192",
}

Requirements

  • python 3.8.8
  • tensorflow-gpu 2.3.0
  • scikit-learn 0.24.1
  • keras 2.4.3
  • networkx 2.5.1
  • tqdm 4.59.0

How to run

To run DAGE on either DBLP or Citeseer-M10, please follow these steps:

  • make sure all the requirements above are satisfied;
  • open the config.py file,
  • edit the dataset_name variable with either "dblp" or "citeseer-m10";
  • configure the hyper-parameters of DAGE or leave them as they are;
  • close the config.py file;
  • execute main.py (i.e., python main.py)