如何解开一个 nlp 模型

Question

我已经训练了一个 spacy 空白的 NER NLP 模型，我想保存它以便在应用程序中使用它，我考虑过 pickle 它我设法使用这两行

import pickle
pickle.dump(nlp, open( "nlp.p", "wb" ))

但是我想用的时候加载不出来我试过了

pickle_in = open('/content/nlp.p', 'rb')
pickle_clf = pickle.load(pickle_in)

它打印

EOFError: Ran out of input

我如何解开它并在文本上测试它

Answer 1

我认为 pickling 应该像往常一样使用 spaCy 模型，但您最好使用内置的序列化方法。

# save your model
nlp.to_disk("my-model") # creates a directory at this path

# load it back
nlp = spacy.load("my-model")

how to unpickle an nlp model