word2vec, sentence2vec, machine reading comprehension, dialog system, text classification, pretrained language model (i.e., XLNet, BERT, ELMo, GPT), sequence labeling, information retrieval, information extraction (i.e., entity, relation and event extraction), knowledge graph, text generation, network embedding

information-retrieval text-classification word2vec text-generation information-extraction knowledge-graph network-embedding sequence-labeling dialogue-systems sentence2vec machine-reading-comprehension pretrained-language-model

Updated Jun 1, 2020
OpenEdge ABL

benedekrozemberczki / SimGNN

Star

A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).

machine-learning deep-learning neural-network tensorflow sklearn pytorch attention-mechanism network-embedding wsdm ged gcn graph-embedding graph-classification tensor-network graph-similarity graph-convolution graph-attention gnn graph-edit-distance simgnn

Updated May 31, 2020
Python

THUDM / GATNE

Star

Source code and dataset for KDD 2019 paper "Representation Learning for Attributed Multiplex Heterogeneous Network"

heterogeneous-network representation-learning multiplex-networks network-embedding attributed-networks

Updated Dec 30, 2019
Python

cnclabs / smore

Star

SMORe: Modularize Graph Embedding for Recommendation

matrix-factorization knowledge-graph representation-learning network-embedding

Updated Feb 9, 2020
C++

Jhy1993 / Representation-Learning-on-Heterogeneous-Graph

Star

Representation-Learning-on-Heterogeneous-Graph

machine-learning deep-learning representation-learning network-embedding graph-embedding heterogeneous-information-networks network-representation-learning graph-neural-network heterogeneous-graph

Updated Mar 13, 2020

benedekrozemberczki / APPNP

Star

A PyTorch implementation of "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019).

machine-learning deep-neural-networks research deep-learning pagerank pytorch deepwalk attention network-embedding gcn iclr node2vec graph-embedding graph-classification node-embedding graph-attention graph-convolutional-neural-networks graph-neural-network ppnp appnp

Updated May 31, 2020
Python

benedekrozemberczki / GEMSEC

Star

The TensorFlow reference implementation of 'GEMSEC: Graph Embedding with Self Clustering' (ASONAM 2019).

machine-learning facebook neural-network clustering tensorflow word2vec community-detection deezer deepwalk matrix-factorization gensim unsupervised-learning embedding network-embedding node2vec graph-embedding gemsec implicit-factorization semisupervised-learning m-nmf

Updated May 31, 2020
Python

thunlp / CANE

Star

Source code and datasets of "CANE: Context-Aware Network Embedding for Relation Modeling"

network-embedding

Updated Mar 30, 2019
Python

benedekrozemberczki / Splitter

Star

A Pytorch implementation of "Splitter: Learning Node Representations that Capture Multiple Social Contexts" (WWW 2019).

machine-learning deep-learning clustering word2vec community-detection pytorch deepwalk gensim factorization network-embedding node2vec graph-embedding overlapping-community-detection deep-neural-network graph-representation-learning node-embedding implicit-factorization graph-neural-network ego-splitting word-vector

Updated Apr 1, 2020
Python

benedekrozemberczki / SEAL-CI

Star

A PyTorch implementation of "Semi-Supervised Graph Classification: A Hierarchical Graph Perspective" (WWW 2019)

deep-learning tensorflow machine-learning-algorithms pytorch deepwalk deep-learning-algorithms network-embedding active-learning graph-convolutional-networks gcn node2vec graph-embedding graph-classification active-learning-module node-classification graph-neural-networks graph-representation-learning graph-convolution gnn graph-neural-network

Updated May 31, 2020
Python

benedekrozemberczki / SGCN

Star

A PyTorch implementation of "Signed Graph Convolutional Network" (ICDM 2018).

Updated May 31, 2020
Python

benedekrozemberczki / datasets

Star

A repository of pretty cool datasets that I collected for network science and machine learning research.

data-science benchmark machine-learning community-detection network-science deepwalk dataset dimensionality-reduction network-analysis network-embedding link-prediction gcn node2vec graph-embedding node-classification graph2vec node-embedding graph-convolution gnn graph-neural-network

Updated May 21, 2020

acbull / pyHGT

Star

Code for "Heterogeneous Graph Transformer" (WWW'20), which is based on pytorch_geometric

transformer heterogeneous-network network-embedding graph-neural-networks graph-representation-learning

Updated Jul 18, 2020
Python

xptree / NetMF

Star

Network Embedding as Matrix Factorization: Unifying DeepWalk, LINE, PTE, and node2vec

social-network deepwalk matrix-factorization social-network-analysis network-embedding node2vec

Updated Jan 27, 2020
Python

xiangyue9607 / BioNEV

Star

Graph Embedding Evaluation / Code and Datasets for "Graph Embedding on Biomedical Networks: Methods, Applications, and Evaluations" (Bioinformatics 2020)