如何在图特征编码中使用PyTorch帧编码

How to use PyTorch frame encoding in graph feature encoding

提问人:SoftwareCo 提问时间:11/15/2023 最后编辑:karelSoftwareCo 更新时间:11/18/2023 访问量:21

问:

我不知道如何在按照 pytorch-geometric 从 csv 加载数据时将 PyTorch 帧编码用于节点特征

我可以简单地将编码类替换为上面链接中定义的编码类,如下所示吗?

from torch_frame import stype
from torch_frame.nn import (
    EmbeddingEncoder,
    FTTransformer,
    LinearEmbeddingEncoder,
    LinearEncoder,
)

movie_x, movie_mapping = load_node_csv(
    movie_path, index_col='movieId', encoders={
        'title': LinearEmbeddingEncoder(),
        'genres': EmbeddingEncoder()
    }
)

我已经尝试了我在问题中提供的代码。它有效,但我不确定它是否正确,或者我是否应该在应用于我的代码之前进行任何操作。请让我知道这是否正确,或者我是否应该对编码器进行任何操作以对我的目的有用。

请指导我是否正确,或者如果不正确,我应该如何在我的情况下使用这些编码?

深度学习 神经网络 pytorch-geometric graph-neural-network

评论


答: 暂无答案