steve's repositories
AutoFIS
AutoFIS: Automatic Feature Interaction Selection in Factorization Models for Click-Through Rate Prediction
bert
TensorFlow code and pre-trained models for BERT
bert-multi-gpu
Feel free to fine tune large BERT models with Multi-GPU and FP16 support.
CogKR
Source code and dataset for paper "Cognitive Knowledge Graph Reasoning for One-shot Relational Learning"
deepwalk
DeepWalk - Deep Learning for Graphs
dual-bandit-kdd-2020
Source code for our paper "Joint Policy-Value Learning for Recommendation" published at KDD 2020.
ESAM
Code for the paper "ESAM: Discriminative Domain Adaptation with Non-Displayed Items to Improve Long-Tail Performance" (SIGIR2020)
Fi_GNN
[CIKM 2019] Code and dataset for "Fi-GNN: Modeling Feature Interactions via Graph Neural Networks for CTR Prediction"
gae
Implementation of Graph Auto-Encoders in TensorFlow
GAT
Graph Attention Networks (https://arxiv.org/abs/1710.10903)
GraphAT
Graph Adversarial Training: Dynamically Regularizing Based on Graph Structure
grec
Future Data Helps Training: Modeling Future Contexts for Session-based Recommendation
HAN
Heterogeneous Graph Neural Network
HierTCN
Hierarchical Temporal Convolutional Networks for Dynamic Recommender Systems
kgpolicy
Reinforced Negative Sampling over Knowledge Graph for Recommendation, WWW2020
LightGCN-parallelized-version
version: parallelize negative sampling on CPU
M2GRL
A demo code of KDD2020 paper "M2GRL: A Multi-task Multi-view Graph Representation Learning Framework for Web-scale Recommender Systems"
MATN
Code for MATN in SIGIR 2020 Industry track
MixText
MixText: Linguistically-Informed Interpolation of Hidden Space for Semi-Supervised Text Classification
MultiObjectiveOptimization
Source code for Neural Information Processing Systems (NeurIPS) 2018 paper "Multi-Task Learning as Multi-Objective Optimization"
neural_graph_collaborative_filtering
Neural Graph Collaborative Filtering, SIGIR2019
Off-Policy-2-Stage
Off-policy Learning in Two-stage Recommender Systems. https://dl.acm.org/doi/pdf/10.1145/3366423.3380130
PRML
PRML algorithms implemented in Python
SDM
Sequential deep matching model for recommender system at Alibaba
SetRank
Implementation of SetRank in SIGIR 2020
solvegp
Sparse Orthogonal Variational Inference for Gaussian Processes (SOLVE-GP)