Skip to content

Commit b633093

Browse files
committed
chore: remove some defaults hyperparameters
1 parent 5045111 commit b633093

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

pyrdf2vec/embedders/word2vec.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ class Word2Vec(Embedder):
1919
_model: The gensim.models.word2vec model.
2020
Defaults to None.
2121
kwargs: The keyword arguments dictionary.
22-
Defaults to { min_count=0, negative=20, vector_size=500 }.
22+
Defaults to { min_count=0 }.
2323
2424
"""
2525

@@ -29,8 +29,6 @@ class Word2Vec(Embedder):
2929
def __init__(self, **kwargs):
3030
self.kwargs = {
3131
"min_count": 0,
32-
"negative": 20,
33-
"vector_size": 500,
3432
**kwargs,
3533
}
3634
self._model = W2V(**self.kwargs)

0 commit comments

Comments
 (0)