create_excel_for_bench.py

import pandas as pd
import os

from prompt_graph.defines import GRAPH_TASKS

graph_dataset_name = GRAPH_TASKS
node_dataset_name = ['PubMed', 'CiteSeer', 'Cora', 'Wisconsin', 'Texas', 'ogbn-arxiv', 'Actor', 'Flickr']
shot_nums = [1,3,5]
for dataset_name in node_dataset_name:
    for shot_num in shot_nums:
        pre_train_types = ['None', 'DGI', 'GraphMAE', 'Edgepred_GPPT', 'Edgepred_Gprompt', 'GraphCL', 'SimGRACE']
        prompt_types = ['None', 'GPPT', 'All-in-one', 'Gprompt', 'GPF', 'GPF-plus']


        column_names = [f"{pre_train}+{prompt}"  for prompt in prompt_types for pre_train in pre_train_types if pre_train != 'None' or prompt == 'None']

        # 创建DataFrame
        data = pd.DataFrame(columns=column_names, index=['Final Accuracy', 'Final F1', 'Final AUROC'])

        gnn_type = 'GCN'

        file_name = gnn_type +"_total_results.xlsx"
        file_path = os.path.join('./Experiment/ExcelResults/Node/'+str(shot_num)+'shot/'+ dataset_name +'/', file_name)
        if not os.path.exists(file_path):
            os.makedirs(os.path.dirname(file_path), exist_ok=True)        
        data.to_excel(file_path)

        # 打印信息确认文件已保存
        print(f"Data saved to {file_path} successfully.")

for dataset_name in graph_dataset_name:
    for shot_num in shot_nums:
        pre_train_types = ['None', 'DGI', 'GraphMAE', 'Edgepred_GPPT', 'Edgepred_Gprompt', 'GraphCL', 'SimGRACE']
        prompt_types = ['None', 'GPPT', 'All-in-one', 'Gprompt', 'GPF', 'GPF-plus']


        column_names = [f"{pre_train}+{prompt}"  for prompt in prompt_types for pre_train in pre_train_types if pre_train != 'None' or prompt == 'None']

        # 创建DataFrame
        data = pd.DataFrame(columns=column_names, index=['Final Accuracy', 'Final F1', 'Final AUROC'])

        gnn_type = 'GCN'

        file_name = gnn_type +"_total_results.xlsx"
        file_path = os.path.join('./Experiment/ExcelResults/Graph/'+str(shot_num)+'shot/'+ dataset_name +'/', file_name)
        if not os.path.exists(file_path):
            os.makedirs(os.path.dirname(file_path), exist_ok=True)        
        data.to_excel(file_path)

        # 打印信息确认文件已保存
        print(f"Data saved to {file_path} successfully.")