Open zzzengzhe opened 1 year ago
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'
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'