alibaba / euler

A distributed graph deep learning framework.
Apache License 2.0
2.89k stars 559 forks source link

euler目前支持导入预训练的embedding向量吗? #331

Open yxz9087 opened 3 years ago

yxz9087 commented 3 years ago

查看了一下源码,想要找到node的embedding table是在哪里定义的,然后用预训练的embedding table替代,但是没有找到。 请问目前euler支持导入预训练的数据吗?

zhangyuhanjc commented 3 years ago

euler实现的embedding层( tf_euler.utils.layers.Embedding)没有直接实现方便的预加载参数,我觉得可以选择以下两种"曲线救国" 1.直接按层名字找到权重加载

image
  1. 我试过可以在euler中穿插使用tf.keras的层, 使用keras的embdding层有参数可以把pretrained_weight直接传入