marius-team / marius

Large scale graph learning on a single machine.
https://marius-project.org
Apache License 2.0
160 stars 45 forks source link

Add postprocessing module for model export #109

Closed rogerwaleffe closed 1 year ago

rogerwaleffe commented 2 years ago

This PR introduces a post processing module which can export trained embeddings to CSV and parquet formats.

The postprocessing module is in-memory only, in the future an out-of-core version will be added.

Contents:

  1. marius_postprocess command
  2. End to end tests of the module
  3. Documentation of usage.