Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/benedekrozemberczki/graph2vec
/ functions
Functions
9 in github.com/benedekrozemberczki/graph2vec
⨍
Functions
9
◇
Types & classes
1
↓ 2 callers
Function
path2name
(path)
src/graph2vec.py:56
↓ 1 callers
Function
dataset_reader
Function to read the graph and features from a json file. :param path: The path to the graph json. :return graph: The graph object. :
src/graph2vec.py:60
↓ 1 callers
Method
do_a_recursion
The method does a single WL recursion. :return new_features: The hash table with extracted WL features.
src/graph2vec.py:32
↓ 1 callers
Method
do_recursions
The method does a series of WL recursions.
src/graph2vec.py:49
↓ 1 callers
Function
main
Main function to read the graph list, extract features. Learn the embedding and save it. :param args: Object with the arguments.
src/graph2vec.py:110
↓ 1 callers
Function
parameter_parser
A method to parse up command line parameters. By default it gives an embedding of the partial NCI1 graph dataset. The default hyperparame
src/param_parser.py:5
↓ 1 callers
Function
save_embedding
Function to save the embedding. :param output_path: Path to the embedding csv. :param model: The embedding model object. :param files
src/graph2vec.py:93
Method
__init__
Initialization method which also executes feature extraction. :param graph: The Nx graph object. :param features: Feature has
src/graph2vec.py:18
Function
feature_extractor
Function to extract WL features from a graph. :param path: The path to the graph json. :param rounds: Number of WL iterations. :retur
src/graph2vec.py:81