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

Fix marius_postprocess and marius_predict bad assumption on general.experiment_name #68

Closed AnzeXie closed 3 years ago

AnzeXie commented 3 years ago

Describe the pull request Made changes to marius_postprocess, marius_predict and related tests, docs on the bad assumption on general.experiment_name. This PR enables marius_postprocess & marius_predict to perform their tasks on directories created by experiments with general.experiment_name other than "marius". Instead of just providing the base directory name (./data/ by default), users are now requried to provide <path.base_directory>/<general.experiment_name> for the command line argument trained_embeddings_directory.

How was this tested A test is added for marius_postprocess & marius_predict. The test exams if marius_postprocess & marius_predict works on embedding directories other than ./data/marius.