wzhut / NESH

Nonparametric Embeddings of Sparse High-Order Interaction Events

This repository contains the code for NESH paper.

If you find this useful, or if you use it in your work, please cite:

  @misc{https://doi.org/10.48550/arxiv.2207.03639,
  doi = {10.48550/ARXIV.2207.03639},
  
  url = {https://arxiv.org/abs/2207.03639},
  
  author = {Wang, Zheng and Xu, Yiming and Tillinghast, Conor and Li, Shibo and Narayan, Akil and Zhe, Shandian},
  
  keywords = {Machine Learning (cs.LG), Machine Learning (stat.ML), FOS: Computer and information sciences, FOS: Computer and information sciences},
  
  title = {Nonparametric Embeddings of Sparse High-Order Interaction Events},
    
  year = {2022},

}

Requirements

Python >= 3.6.0
Pytorch >= 1.10.1
numpy >= 1.15.4
Matlab >= 2018b