Function build_args

code/utils.py:38–120

(task=None)

# ======================================================================

def build_args(task=None):

    parser = argparse.ArgumentParser(description='UGAD')
    # General settings
    parser.add_argument("--strategy", type=str, default="graphinfomax", help="Pretrain model strategy")
    parser.add_argument("--kernel", type=str, default="gcn", help="GNN model type")
    parser.add_argument("--dataset", type=str, default="Cora", help="Dataset for this model")
    parser.add_argument("--data_dir", type=str, default="./datasets/", help="Data directory")
    parser.add_argument("--model_dir", type=str, default="./ckpts/", help="Folder to save model")
    parser.add_argument("--log_dir", type=str, default="./logs/", help="Folder to save logger")

    # Model Configuration settings
    parser.add_argument("--seed", type=int, nargs="+", default=[12], help="Random seed")
    parser.add_argument("--hid_dim", type=int, default=768, help="Hidden layer dimension")
    parser.add_argument("--num_layer", type=int, default=5, help="Number of hidden layer in main model")
    parser.add_argument("--act", type=str, default='relu', help="Activation function type")
    parser.add_argument("--norm", type=str, default="", help="Normlaization layer type")
    parser.add_argument("--linear_layer", type=int, default=2, help="Number of linear layer in prediction model")
    parser.add_argument("--mask_ratio", type=float, default=0.5, help="Masking ratio for GraphMAE")
    parser.add_argument("--replace_ratio", type=float, default=0, help="Replace ratio for GraphMAE")
    # Dataset settings
    parser.add_argument("--unify", action="store_true", default=False, help="SVD unify feature dimension")
    parser.add_argument("--unify_dim", type=int, default=100, help="SVD reduction dimension")
    parser.add_argument("--aug", type=str, default="dnodes")

    # Training settings
    parser.add_argument("--epoch", type=int, default=1000, help="The max number of epochs")
    parser.add_argument("--lr", type=float, default=0.001, help="Learning rate of optimizer")
    parser.add_argument("--l2", type=float, default=0, help="Coefficient of L2 penalty")
    parser.add_argument("--decay_rate", type=float, default=1, help="Decay rate of learning rate")
    parser.add_argument("--decay_step", type=int, default=100, help="Decay step of learning rate")
    parser.add_argument("--eval_epoch", type=int, default=1, help="Number of evaluation epoch")
    parser.add_argument("--sparse", action='store_true', default=False, help="Indicator of sparse computation")
    parser.add_argument("--down_epoch", type=int, default=50, help="The max number of epochs for finetune")
    parser.add_argument("--contrast_batch", type=int, default=256, help="Batch size for contrastive learning")
    parser.add_argument("--patience", type=int, default=800, help="Early stop patience for pretraining")

    # Hyperparameters
    parser.add_argument("--norm_type", type=str, default='sym', help="Type of normalization of adjacency matrix")
    parser.add_argument("--dropout", type=float, default=0, help="Dropout rate for node in training")
    parser.add_argument("--edge_dropout", type=float, default=0, help="Dropout rate for edge in training")

    # Auxiliary
    parser.add_argument("--save_model", action='store_true', default=False, help="Indicator to save trained model")
    parser.add_argument("--load_model", action='store_true', default=False, help="Indicator to load trained model")
    parser.add_argument("--log", action='store_true', default=False, help="Indicator to write logger file")
    parser.add_argument("--use_cfg", action="store_true", default=False, help="Indicator to use best configurations")

    # GPU settings
    parser.add_argument("--no_cuda", action='store_true', default=False, help="Indicator of GPU availability")
    parser.add_argument("--device", type=int, default=0, help='Which gpu to use if any')

    # Text settings
    parser.add_argument("--if_text", action='store_true', help="Indicator of text-enhanced dataset")
    parser.add_argument("--cl", action='store_true', help="Indicator of contrastive learning")
    parser.add_argument("--load_pkl", action='store_true', help="Indicator of loading from pretrained")

    parser.add_argument("--text_encoder", type=str, default='SentenceBert', help="Text encoder type")