Skip to content

我在用自己的数据执行的时候报这个错误,不知道是什么原因,所以想请教一下 #126

@zzzengzhe

Description

@zzzengzhe

Generating neighbors for layer 0
0%| | 192/2136051 [00:00<?, ?it/s]
Traceback (most recent call last):
File "src/main_pytorch.py", line 287, in
average_auc, average_f1, average_pr = train_model(training_data_by_type, feature_dic)
File "src/main_pytorch.py", line 158, in train_model
neighbors = generate_neighbors(network_data, vocab, num_nodes, edge_types, neighbor_samples)
File "D:\GATNE-master\src\utils.py", line 241, in generate_neighbors
iy = vocab[y].index
KeyError: '65787'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions