How to get the embedding of the entities and relations?

JD-AI-Research-Silicon-Valley / SACN

End-to-end Structure-Aware Convolutional Networks for Knowledge Base Completion

MIT License

113 stars 30 forks source link

Open Danga001 opened 3 years ago

Danga001 commented 3 years ago

Hello, I want to get the embedding of the entitys and relations during the training process of SACN and other classic models

chaoshangcs commented 3 years ago

Hi, thanks for your question. You can output the entity and relation embeddings on both training and evaluation in models.py.