snap-stanford / ipf-network-inference

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

IPF for dynamic network inference

This respository contains code to run all experiments for "Inferring dynamic networks from marginals with iterative proportional fitting". The main files are:

  • test_ipf.py: functions to run IPF, test and enable IPF convergence (our ConvIPF algorithm), and run the equivalent Poisson regression.
  • experiments_with_data.py: functions to run experiments with data - synthetic data, SafeGraph mobility data, and CitiBike bikeshare data
  • icml_experiments.ipynb: notebook to recreate all results and figures reported in the paper.

The other files are imported from other repos (Chang et al., 2021) to use SafeGraph mobility data.

About


Languages

Language:Jupyter Notebook 78.0%Language:Python 22.0%