Closed Demirrr closed 3 years ago
It's printed and logged whenever you train a model. For the published models, run
kge dump checkpoint fb15k-237-rescal.pt --keys valid_trace | grep num_parameters | head -1
to extract it. Is this what you mean?
I am afraid something is incomplete.
All steps in quick-start are completed, i.e., the script of toy example was successfully executed.
Executing kge dump checkpoint fb15k-237-rescal.pt --keys valid_trace | grep num_parameters | head -1
throwed the following exception Exception: Could not find checkpoint in fb15k-237-rescal.pt
My main goal is to fully make use of the results reported in "YOU CAN TEACH AN OLD DOG NEW TRICKS! ON TRAINING KNOWLEDGE GRAPH EMBEDDINGS". More specifically, I am interesting in the number of trainable parameters for all models in the Table2. Apart from this, I was wondering whether you could name one or two frameworks that remove unseen entities from the test set.
Cheers
You need to download the respective checkpoint from the KGE homepage first. The checkpoint links are in the "pretrained model" column.
It worked like a charm. Thanks!
Hey,
thank you for great work! I was wondering whether the number of parameters of models --including RESCAL, ComplEx and ConvE have been reported in somewhere.
Cheers, Caglar Demir