For a single network of 17804 nodes and 308081 edges, _create_pyg_graphs execution time was reduced from 2614 seconds to 0.75 seconds - an almost 3500x speedup.
Adds better support for loading pretrained models (#36)
Exposes neighborhood sample size to the config as neighbor_sample_size key
Enables saving of training loss data to a .tsv file (#40)
_create_pyg_graphs
execution time was reduced from 2614 seconds to 0.75 seconds - an almost 3500x speedup.neighbor_sample_size
key