()
| 72 | |
| 73 | |
| 74 | def build_args(): |
| 75 | parser = argparse.ArgumentParser(description='UGAD') |
| 76 | # General settings |
| 77 | parser.add_argument("--strategy", type=str, |
| 78 | default="graphinfomax", help="Pretrain model strategy") |
| 79 | parser.add_argument("--kernel", type=str, |
| 80 | default="gcn", help="GNN model type") |
| 81 | parser.add_argument("--dataset", type=str, nargs='+', |
| 82 | default=["Cora"], help="Pre-train datasets for this model") |
| 83 | parser.add_argument("--data_dir", type=str, |
| 84 | default="./datasets/", help="Data directory") |
| 85 | parser.add_argument("--model_dir", type=str, |
| 86 | default="./ckpts/", help="Folder to save model") |
| 87 | parser.add_argument("--log_dir", type=str, |
| 88 | default="./logs/", help="Folder to save logger") |
| 89 | |
| 90 | # Model Configuration settings |
| 91 | parser.add_argument("--seed", type=int, nargs="+", |
| 92 | default=[12], help="Random seed") |
| 93 | parser.add_argument("--hid_dim", type=int, default=768, |
| 94 | help="Hidden layer dimension") |
| 95 | parser.add_argument("--num_layer", type=int, default=5, |
| 96 | help="Number of hidden layer in main model") |
| 97 | parser.add_argument("--act", type=str, default='relu', |
| 98 | help="Activation function type") |
| 99 | parser.add_argument("--norm", type=str, default="", |
| 100 | help="Normlaization layer type") |
| 101 | parser.add_argument("--linear_layer", type=int, default=2, |
| 102 | help="Number of linear layer in prediction model") |
| 103 | parser.add_argument("--mask_ratio", type=float, |
| 104 | default=0.5, help="Masking ratio for GraphMAE") |
| 105 | parser.add_argument("--replace_ratio", type=float, |
| 106 | default=0, help="Replace ratio for GraphMAE") |
| 107 | parser.add_argument("--decay_rate", type=float, default=1, |
| 108 | help="Decay rate of learning rate") |
| 109 | parser.add_argument("--decay_step", type=int, default=100, |
| 110 | help="Decay step of learning rate") |
| 111 | parser.add_argument("--lr", type=float, default=0.0001, |
| 112 | help="Learning rate of optimizer") |
| 113 | parser.add_argument("--gradient_accumulation_steps", |
| 114 | type=int, default=4, help="gradient accumulation steps") |
| 115 | parser.add_argument("--k", type=int, default=2, help="k-hop subgraph") |
| 116 | |
| 117 | # Dataset settings |
| 118 | parser.add_argument("--unify", action="store_true", |
| 119 | default=False, help="SVD unify feature dimension") |
| 120 | parser.add_argument("--unify_dim", type=int, default=100, |
| 121 | help="SVD reduction dimension") |
| 122 | parser.add_argument("--aug", type=str, default="dnodes") |
| 123 | |
| 124 | parser.add_argument("--test_dataset", type=str, nargs='+', |
| 125 | default=["Cora"], help="Pre-train datasets for this model") |
| 126 | parser.add_argument("--datasetnorm", action='store_true', |
| 127 | help="Data normalization") |
| 128 | # Training settings |
| 129 | parser.add_argument("--epoch", type=int, default=10, |
| 130 | help="The max number of epochs") |
| 131 | parser.add_argument("--if_norm", action='store_true', |
no outgoing calls
no test coverage detected