Skip to content

Cell FLOWer: Inferring Cell Complexes from Edge Flows

License

Notifications You must be signed in to change notification settings

josefhoppe/cell-flower

Repository files navigation

Cell FLOWer

Published at Learning on Graphs Conference 2023 Best Paper at LoG 2023 - LoG Conference on Twitter arXiv:2309.01632 DOI:10.18154/RWTH-2024-03864 MIT license Python version Build status Package version on PyPI

Cell FLOWer processes edge flows using cell complexes. It was developed for the paper Representing Edge Flows on Graphs via Sparse Cell Complexes. You can find the evaluation workflow here. Install it using:

pip install cell-flower

How to use it (Also check out the complete examples):

import cell_flower as cf
import networkx as nx

G = ... # nx.Graph
CC, node_list, node_index_map = cf.nx_graph_to_cc(G) # converts nodes in G to int and also returns the mapping
flows = ... # np.ndarray, shape (samples, edges)

CC_prime = cf.cell_inference_approximation(CC, flows, 2, 2, n_clusters=5)
# Check to see the cells that were recovered; map back to node labels
[ cf.index_to_cell(cell, node_list) for cell in CC_prime.get_cells(2) ]

If you use Cell FLOWer, please cite the following paper:

@InProceedings{hoppe2024representing,
  title = 	 {Representing Edge Flows on Graphs via Sparse Cell Complexes},
  author =       {Hoppe, Josef and Schaub, Michael T},
  booktitle = 	 {Proceedings of the Second Learning on Graphs Conference},
  pages = 	 {1:1--1:22},
  year = 	 {2024},
  editor = 	 {Villar, Soledad and Chamberlain, Benjamin},
  volume = 	 {231},
  series = 	 {Proceedings of Machine Learning Research},
  month = 	 {27--30 Nov},
  publisher =    {PMLR},
  pdf = 	 {https://proceedings.mlr.press/v231/hoppe24a/hoppe24a.pdf},
  url = 	 {https://proceedings.mlr.press/v231/hoppe24a.html},
}

Acknowledgements

Funded by the European Union (ERC, HIGH-HOPeS, 101039827). Views and opinions expressed are however those of the author(s) only and do not necessarily reflect those of the European Union or the European Research Council Executive Agency. Neither the European Union nor the granting authority can be held responsible for them.

About

Cell FLOWer: Inferring Cell Complexes from Edge Flows

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages