attack.py

import os
import sys
import torch
from config.config import get_config
from utils.hyperattack import RandomAttack_hyperedges

def main():
    cfg=get_config()
    
    print(f"==================================================")
    print(f"         HGNN-Shield Attack Generator")
    print(f"==================================================")
    print(f"Dataset:  {cfg.data.dataset}")
    print(f"DataRoot: {cfg.data.root}")
    print(f"Mode:     {cfg.attack.mode}")
    print(f"Rate:     {cfg.attack.rate}")
    print(f"Device:   {cfg.data.device}")
    print(f"--------------------------------------------------")

    try:
        result = RandomAttack_hyperedges(cfg, cfg.data.dataset)

        print(f"[Success] Attack generation completed.")
    except Exception as e:
        print(f"[Error] Attack generation failed: {e}")
        import traceback
        traceback.print_exc()

if __name__ == '__main__':
    main()