-
Notifications
You must be signed in to change notification settings - Fork 47
/
Copy pathrun_deep_graph_contrastive.sh
executable file
·61 lines (54 loc) · 2.14 KB
/
run_deep_graph_contrastive.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#! /bin/bash
#
# Tencent is pleased to support the open source community by making embedx
# available.
#
# Copyright (C) 2021 THL A29 Limited, a Tencent company. All rights reserved.
#
# Licensed under the BSD 3-Clause License and other third-party components,
# please refer to LICENSE for details.
#
# Author: Chunchen Su ([email protected])
#
set -e
cd "$(dirname "$0")"
source runtime.sh
readonly DATASET="cora"
readonly DATASET_DIR="${DEMO_DIR}/data/${DATASET}"
readonly TRAIN_LABELS="${DATASET_DIR}/train_labels"
readonly TEST_LABELS="${DATASET_DIR}/test_labels"
# graph flags
readonly FLAGS_node_graph="${DATASET_DIR}/context"
readonly FLAGS_node_feature="${DATASET_DIR}/node_feature"
readonly FLAGS_gs_thread_num=8
# trainer & predictor flags
readonly FLAGS_thread_num=8
readonly FLAGS_model="deep_graph_contrastive"
readonly FLAGS_model_config="config=0:54:64;depth=2;dim=128;alpha=0.1;sparse=1;use_neigh_feat=0;tau=0.4"
readonly FLAGS_instance_reader="deep_graph_contrastive_inst_reader"
readonly FLAGS_optimizer="adam"
readonly FLAGS_optimizer_config="rho1=0.9;rho2=0.999;alpha=5e-4;beta=1e-8"
readonly FLAGS_model_shard=10
readonly FLAGS_in="${FLAGS_node_graph}"
readonly FLAGS_epoch=200
readonly FLAGS_out_model="model"
readonly FLAGS_out_predict="embedding"
################################################################
# Train
################################################################
FLAGS_instance_reader_config="num_neighbors=50,50;use_neigh_feat=0;left_edge_drop_prob=0.1;right_edge_drop_prob=0.3;left_feat_mask_prob=0.2;right_feat_mask_prob=0.4"
FLAGS_batch=2708
FLAGS_target_type=0
run_trainer ${DATASET}
################################################################
# Predict
################################################################
FLAGS_instance_reader_config="num_neighbors=50,50;use_neigh_feat=0;is_train=0"
FLAGS_in_model="${FLAGS_out_model}"
FLAGS_batch=1
FLAGS_target_type=2
run_predictor ${DATASET}
################################################################
# Evaluate
################################################################
evaluate_embedding "${TEST_LABELS}" "${TRAIN_LABELS}" "${FLAGS_out_predict}" "SGD"