Closed ditordccaa closed 3 years ago
After some verification, there is no problem with the package it's more of the scripts provided for training a custom model. Specifically the script 05_export.py.
I made a change in line 146 by wrapping all_senese with list like the following:
s2v = Sense2Vec(shape=(n_vectors, vector_size), senses=list(all_senses))
I hope it helps.
Hi there, I'm trying to save a sense2vec model to disk using the to_disk method but it fails with an error.
I tried to track down the issue by looking to the ujson folder of the srsly package but couldn't understand since I believe it's wirtten in C and I don't have any knowledge for that :)
While I was experimenting, I've noticed when I print s2v.cfg, it's a dictionary in the form below and I think that the value for the keyword 'senses' should be a list and not a dictionary.
Python version
Python 3.6.9
Packages versions
srsly 1.0.4
spacy 2.2.4
catalogue 1.0.0
numpy 1.18.5