()
| 111 | |
| 112 | |
| 113 | def main(): |
| 114 | args = _parse_args() |
| 115 | node_embedding = _load_data_as_dict(args.embedding_file, dtype=float) |
| 116 | train_node_label = _load_data_as_dict(args.train_label_file, dtype=int) |
| 117 | test_node_label = _load_data_as_dict(args.test_label_file, dtype=int) |
| 118 | |
| 119 | train_embeddings = [] |
| 120 | train_labels = [] |
| 121 | for node in train_node_label: |
| 122 | assert node in node_embedding |
| 123 | train_embeddings.append(node_embedding[node]) |
| 124 | train_labels.append(train_node_label[node]) |
| 125 | |
| 126 | test_embeddings = [] |
| 127 | test_labels = [] |
| 128 | for node in test_node_label: |
| 129 | assert node in node_embedding |
| 130 | test_embeddings.append(node_embedding[node]) |
| 131 | test_labels.append(test_node_label[node]) |
| 132 | |
| 133 | logging.info("Running regression...") |
| 134 | _train_and_evaluate(np.array(train_embeddings), np.array(train_labels), |
| 135 | np.array(test_embeddings), np.array(test_labels), |
| 136 | args.average, args.classifier_name) |
| 137 | |
| 138 | |
| 139 | if __name__ == '__main__': |
no test coverage detected