Skip to content

预训练步骤 #28

@db-beep

Description

@db-beep

作者您好,我在试图将SDCN框架运用到我自己的数据集上遇到一些问题,想向您请教。(无标签+非graph数据)
根据您之前说的将SDCN运用到其他数据集的三个步骤,第一步,创建自己数据集的KNN图,这一步我已经完成。第二步预训练步骤,我所输入的x的shape为[8500,20000],然后根据代码,需要修改model=AE{}这一部分的代码,在CPU/GPU上训练都提示大小不匹配的问题,即使将预训练的网络修改成shape相同的,同样还是无法训练。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions