DeepGraphLearning / graphvite

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

KnowledgeGraphSolver Error #107

Open DaliaDawod opened 1 year ago

DaliaDawod commented 1 year ago

I have this error. Can you help me to fix it? I saw on documentation that dim is dim: 32, 64, 96, 128, 256, 512 but my entity embedding is 768,

  File "/opt/conda/lib/python3.8/site-packages/graphvite/helper.py", line 105, in __new__
    raise AttributeError("Can't find an instantiation of %s with %s" % (cls.name, ", ".join(strings)))
AttributeError: Can't find an instantiation of KnowledgeGraphSolver with dim=768, float_type=dtype.float32, index_type=dtype.uint32

I tried reshaping to 512 but I get

100%|██████████| 828/828 [00:00<00:00, 788933.15it/s]loading knowledge graph from /repo/kepler/wikidata5m_transductive_train.txt
4.04445e-06%
48.5112%
97.0195%
<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
KnowledgeGraph<uint32>
------------------ Graph -------------------
#entity: 4594149, #relation: 822
#triplet: 20614279, normalization: no
>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
[time] KnowledgeGraphApplication.load: 21.4308 s
Killed

I am not sure I am using the correct entity2id and relation2id as I tried to produce them but I do not know if it is the correct way.

How can I solve this?