Update README.md

2019-02-14 21:42:28 +08:00 · 2019-02-14 21:42:28 +08:00 · 1f3438762f
commit 1f3438762f
parent a804ec8343
1 changed files with 1 additions and 1 deletions
--- a/README.md
+++ b/README.md
@ -72,7 +72,7 @@ embeddings = model.get_embeddings()# get embedding vectors
 ```python
 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
 embeddings = model.get_embeddings()# get embedding vectors
 ```