Skip to content
This repository was archived by the owner on Jan 22, 2023. It is now read-only.

Update features.py for GenSim >= 4 #16

Open
tonyreina wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Update features.py for GenSim >= 4 #16

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
df76586
Update features.py
tonyreina May 28, 2021
abd771e
Update features.py
tonyreina May 29, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion mol2vec/features.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,9 @@
import timeit
from joblib import Parallel, delayed

import gensim

GENSIM4 = gensim.__version__ > "4" # GENSIM4 has some API changes from GENSIM 3

class DfVec(object):
"""
Expand Down Expand Up @@ -422,7 +425,10 @@ def sentences2vec(sentences, model, unseen=None):
-------
np.array
"""
keys = set(model.wv.vocab.keys())
if GENSIM4:
keys = set(model.wv.key_to_index)
else:
keys = set(model.wv.vocab.keys())
vec = []
if unseen:
unseen_vec = model.wv.word_vec(unseen)
Expand Down