DeepGraphLearning / graphvite

GraphVite: A General and High-performance Graph Embedding System
https://graphvite.io
Apache License 2.0
1.21k stars 151 forks source link

Error when loading pre-trained Wikidata5M models (.pkl files) #99

Closed lena-schwert closed 2 years ago

lena-schwert commented 2 years ago

Hey, I want to use the Wikidata5M pretrained embeddings from here for my project.

However I get an error when trying to load the model with pickle like you suggest it:

import pickle
with open("transe_wikidata5m.pkl", "rb") as fin:
    model = pickle.load(fin)

This code produces an error: ModuleNotFoundError: No module named 'easydict'

I am using a Python 3.8.12 conda environment.

Can you give me any hints on how to resolve this? Please let me know when you need additional details.

lena-schwert commented 2 years ago

Ok the solution, is simple, just install easydict with conda install -c conda-forge easydict.