Graph_algorithm/GraphEmbedding
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Weichen Shen 2019-02-14 21:42:28 +08:00
parent a804ec8343
commit 1f3438762f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -72,7 +72,7 @@ embeddings = model.get_embeddings()# get embedding vectors
```python ```python
G = nx.read_edgelist('../data/flight/brazil-airports.edgelist',create_using=nx.DiGraph(),nodetype=None,data=[('weight',int)])#read graph G = nx.read_edgelist('../data/flight/brazil-airports.edgelist',create_using=nx.DiGraph(),nodetype=None,data=[('weight',int)])#read graph
model = SDNE(G,hidden_size=[256,128]) #init model model = model = Struc2Vec(G, 10, 80, workers=4, verbose=40, ) #init model
model.train(batch_size=3000,epochs=40,verbose=2)# train model model.train(batch_size=3000,epochs=40,verbose=2)# train model
embeddings = model.get_embeddings()# get embedding vectors embeddings = model.get_embeddings()# get embedding vectors
``` ```