Skip to content

Commit

Permalink
Mage Node2Vec update after bumping Gensim version (#967)
Browse files Browse the repository at this point in the history
* update node2vec

* Wording changes
  • Loading branch information
imilinovic authored Aug 14, 2024
1 parent 4ac6a1b commit 5cd1951
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pages/advanced-algorithms/available-algorithms/node2vec.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -118,10 +118,10 @@ The procedure returns calculated embeddings and the corresponding list of embedd
training progresses.
- `sg : {0, 1}` ➡ Training algorithm: 1 for skip-gram; otherwise CBOW.
- `hs : {0, 1}` ➡ If 1, hierarchical softmax will be used for model training. If
0, and `negative` is non-zero, negative sampling will be used.
0, and `negative` is > 0, negative sampling will be used.
- `negative : integer` ➡ If > 0, negative sampling will be used, the integer for
negative specifies how many "noise words" should be drawn (usually
between 5-20). If set to 0, no negative sampling is used.
between 5-20). If set to 0, no negative sampling is used. Either `negative` must be > 0 or `hs` must be set to 1.
- `epochs : integer (default=5)`
- `edge_weight_property : string (default="weight")`

Expand Down

0 comments on commit 5cd1951

Please sign in to comment.