Skip to content

Commit

Permalink
minor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
AnzeXie committed Aug 26, 2021
1 parent 3dcd491 commit 54207de
Showing 1 changed file with 2 additions and 8 deletions.
10 changes: 2 additions & 8 deletions src/python/tools/preprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,12 +27,6 @@
from marius.tools.config_generator import DEFAULT_CONFIG_FILE
from marius.tools.csv_converter import general_parser

# REMEMBER TO CHANGE THE DOCUMENTS ON OTUPUT_DIRECTORY






def live_journal(data_dir, num_partitions=1, split=(.05, .05)):
"""Preprocesses the dataset live_journal.
Expand Down Expand Up @@ -400,9 +394,9 @@ def codex_m(data_dir, num_partitions=1):


def codex_l(data_dir, num_partitions=1):
"""Preprocesses the dataset codex_l.
"""Preprocesses the dataset codex_l.
During preprocessing, Marius has randomly assigned integer ids to each node
During preprocessing, Marius has randomly assigned integer ids to each node
and edge_type, where the mappings to the original ids are stored in
node_mapping.txt and rel_mapping.txt.
The edge list in original dataset files is then converted to an [|E|, 3]
Expand Down

0 comments on commit 54207de

Please sign in to comment.