Fatemeh Salehi Rizi and Michael Granitzer. "Multi-task Network Embedding with Adaptive Loss Weighting." In 2020 IEEE/ACM International Conference on Advances in Social Networks Analysis and Mining (ASONAM), IEEE, 2020.
- To generate embeddings run main.py
- main.py: generates embeddings
- loss_weighting: it's an adaptive loss weighting layer in neural networks
- link_pred: link prediction
- classify: node classification
- att_infer: attribute prediction
- Cora: Cora Dataset
- Citeseer: Citeseer Dataset
- PubMed: PubMed Dataset
- results: contains result files
- loss: contains loss values and task weights
The presented work was developed within the Provenance Analytics project funded by the German Federal Ministry of Education and Research, grant agreement number 03PSIPT5C.