/data/home/weiyb/miniconda3/envs/NLGraph/lib/python3.10/site-packages/torchvision/models/_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead.
  warnings.warn(
/data/home/weiyb/miniconda3/envs/NLGraph/lib/python3.10/site-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=ResNet50_Weights.IMAGENET1K_V1`. You can also use `weights=ResNet50_Weights.DEFAULT` to get the most up-to-date weights.
  warnings.warn(msg)
cat_node_feat_mf:  False
cat_n2v_feat:  False
Namespace(use_val_edges_as_input=False, data_name='cora', neg_mode='equal', gnn_model='GCN', score_model='mlp_score', load=None, loadmod=False, savemod=False, num_layers=1, num_layers_predictor=3, hidden_channels=128, dropout=0.3, batch_size=256, lr=0.01, epochs=1000, eval_steps=5, runs=5, kill_cnt=10, output_dir='/data/home/weiyb/HeaRT/checkpoints/GNN', l2=1e-07, seed=42, velr=0.01, fslr=0.01, hop_num=2, id=0, feat=10, VE='resnet50_tuned', test=False, layout_aug=False, save=False, use_saved_model=False, metric='MRR', device=0, log_steps=1, gin_mlp_layer=2, gat_head=1, cat_node_feat_mf=False, cat_n2v_feat=False)
the number of nodes in cora is:  2708
2708 tensor(2707)
dataset split 
train edge 4488
valid edge 263
valid edge_neg 263
test edge 527
test edge_neg 527
#################################           0           #################################
seed:  0
No saved model parameters found. Starting training from scratch.
layers in gcn:  1
train valid_pos valid_neg test_pos test_neg torch.Size([263]) torch.Size([263]) torch.Size([263]) torch.Size([527]) torch.Size([527])
Hits@1
2024-06-26 06:28:02,160 - [INFO] - Run: 01, Epoch: 05, Loss: 1.1242, Valid: 7.60%, Test: 0.76%
Hits@3
2024-06-26 06:28:02,160 - [INFO] - Run: 01, Epoch: 05, Loss: 1.1242, Valid: 11.41%, Test: 6.07%
Hits@10
2024-06-26 06:28:02,160 - [INFO] - Run: 01, Epoch: 05, Loss: 1.1242, Valid: 22.81%, Test: 17.46%
Hits@100
2024-06-26 06:28:02,160 - [INFO] - Run: 01, Epoch: 05, Loss: 1.1242, Valid: 62.74%, Test: 51.04%
MRR
2024-06-26 06:28:02,160 - [INFO] - Run: 01, Epoch: 05, Loss: 1.1242, Valid: 12.60%, Test: 6.04%
AUC
2024-06-26 06:28:02,160 - [INFO] - Run: 01, Epoch: 05, Loss: 1.1242, Valid: 65.81%, Test: 69.08%
AP
2024-06-26 06:28:02,160 - [INFO] - Run: 01, Epoch: 05, Loss: 1.1242, Valid: 70.68%, Test: 72.05%
---
Best Current: 6.04_5
train valid_pos valid_neg test_pos test_neg torch.Size([263]) torch.Size([263]) torch.Size([263]) torch.Size([527]) torch.Size([527])
Hits@1
2024-06-26 09:11:12,563 - [INFO] - Run: 01, Epoch: 10, Loss: 0.3502, Valid: 0.00%, Test: 0.00%
Hits@3
2024-06-26 09:11:12,564 - [INFO] - Run: 01, Epoch: 10, Loss: 0.3502, Valid: 0.00%, Test: 0.00%
Hits@10
2024-06-26 09:11:12,564 - [INFO] - Run: 01, Epoch: 10, Loss: 0.3502, Valid: 0.00%, Test: 0.00%
Hits@100
2024-06-26 09:11:12,564 - [INFO] - Run: 01, Epoch: 10, Loss: 0.3502, Valid: 56.27%, Test: 32.45%
MRR
2024-06-26 09:11:12,564 - [INFO] - Run: 01, Epoch: 10, Loss: 0.3502, Valid: 1.89%, Test: 0.89%
AUC
2024-06-26 09:11:12,564 - [INFO] - Run: 01, Epoch: 10, Loss: 0.3502, Valid: 60.02%, Test: 58.42%
AP
2024-06-26 09:11:12,564 - [INFO] - Run: 01, Epoch: 10, Loss: 0.3502, Valid: 57.16%, Test: 55.43%
