Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel
Resources  
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Explore Pricing
Product  
Data Engine LLMs Platform Enterprise
Connect to Dagshub Discord channel
Register
Login

Simon
/
graph-barlow-twins
mirror of https://github.com/pbielak/graph-barlow-twins

Watch 1
Issues   Discussions 0

dvc.yaml 5.5 KB
Permalink History Raw

You have to be logged in to leave a comment. Sign In
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209 
  1. stages:
    2. 

  2.   preprocess_dataset:
    3. 

  3.     foreach:
    4. 

  4.       - WikiCS
    5. 

  5.       - Amazon-CS
    6. 

  6.       - Amazon-Photo
    7. 

  7.       - Coauthor-CS
    8. 

  8.       - Coauthor-Physics
    9. 

  9.       - ogbn-arxiv
    10. 

  10.       - PPI
    11. 

  11.       - ogbn-products
    12. 

  12.     do:
    13. 

  13.       wdir: .
    14. 

  14.       cmd: PYTHONPATH=. python3 experiments/scripts/preprocess_dataset.py ${item}
    15. 

  15.       deps:
    16. 

  16.         - experiments/scripts/preprocess_dataset.py
    17. 

  17.       outs:
    18. 

  18.         - data/datasets/${item}/
    19. 

  19. 

  20.   ### Full batch scenario
    21. 

  21.   full_batch_hps:
    22. 

  22.     foreach:
    23. 

  23.       - WikiCS
    24. 

  24.       - Amazon-CS
    25. 

  25.       - Amazon-Photo
    26. 

  26.       - Coauthor-CS
    27. 

  27.       - Coauthor-Physics
    28. 

  28.       - ogbn-arxiv
    29. 

  29.       - PPI
    30. 

  30.     do:
    31. 

  31.       wdir: .
    32. 

  32.       cmd: PYTHONPATH=. python3 experiments/scripts/full_batch/hps.py ${item}
    33. 

  33.       params:
    34. 

  34.         - experiments/configs/full_batch/hps.yaml:
    35. 

  35.             - ${item}
    36. 

  36.       deps:
    37. 

  37.         - experiments/scripts/full_batch/hps.py
    38. 

  38.         - data/datasets/${item}/
    39. 

  39.       outs:
    40. 

  40.         - data/full_batch/hps/${item}/log.csv
    41. 

  41.         - data/full_batch/hps/${item}/best.json
    42. 

  42. 

  43.   full_batch_train:
    44. 

  44.     foreach:
    45. 

  45.       - WikiCS
    46. 

  46.       - Amazon-CS
    47. 

  47.       - Amazon-Photo
    48. 

  48.       - Coauthor-CS
    49. 

  49.       - Coauthor-Physics
    50. 

  50.       - ogbn-arxiv
    51. 

  51.       - PPI
    52. 

  52.     do:
    53. 

  53.       wdir: .
    54. 

  54.       cmd: PYTHONPATH=. python3 experiments/scripts/full_batch/train.py ${item}
    55. 

  55.       params:
    56. 

  56.         - experiments/configs/full_batch/train.yaml:
    57. 

  57.             - ${item}
    58. 

  58.         - data/full_batch/hps/${item}/best.json:
    59. 

  59.             - SAME
    60. 

  60.       deps:
    61. 

  61.         - experiments/scripts/full_batch/train.py
    62. 

  62.         - gssl/full_batch/model.py
    63. 

  63.         - data/datasets/${item}/
    64. 

  64.       outs:
    65. 

  65.         - data/full_batch/train/${item}/
    66. 

  66. 

  67.   ### Batched scenario
    68. 

  68.   batched_train:
    69. 

  69.     foreach:
    70. 

  70.       - WikiCS
    71. 

  71.       - Amazon-CS
    72. 

  72.       - Amazon-Photo
    73. 

  73.       - Coauthor-CS
    74. 

  74.       - Coauthor-Physics
    75. 

  75.       - ogbn-arxiv
    76. 

  76.       - PPI
    77. 

  77.     do:
    78. 

  78.       wdir: .
    79. 

  79.       cmd: PYTHONPATH=. python3 experiments/scripts/batched/train.py ${item}
    80. 

  80.       params:
    81. 

  81.         - experiments/configs/batched/train.yaml:
    82. 

  82.             - ${item}
    83. 

  83.         - data/full_batch/hps/${item}/best.json:
    84. 

  84.             - SAME
    85. 

  85.       deps:
    86. 

  86.         - experiments/scripts/batched/train.py
    87. 

  87.         - data/datasets/${item}/
    88. 

  88.       outs:
    89. 

  89.         - data/batched/train/${item}/
    90. 

  90. 

  91.   ### OGBn-Products
    92. 

  92.   batched_hps_ogbn_products:
    93. 

  93.     wdir: .
    94. 

  94.     cmd: PYTHONPATH=. python3 experiments/scripts/batched/hps.py ogbn-products
    95. 

  95.     params:
    96. 

  96.       - experiments/configs/batched/hps.yaml:
    97. 

  97.           - ogbn-products
    98. 

  98.     deps:
    99. 

  99.       - experiments/scripts/batched/hps.py
    100. 

  100.       - data/datasets/ogbn-products/
    101. 

  101.     outs:
    102. 

  102.       - data/batched/hps/ogbn-products/log.csv
    103. 

  103.       - data/batched/hps/ogbn-products/best.json
    104. 

  104. 

  105.   batched_train_ogbn_products:
    106. 

  106.     wdir: .
    107. 

  107.     cmd: PYTHONPATH=. python3 experiments/scripts/batched/train.py ogbn-products
    108. 

  108.     params:
    109. 

  109.       - experiments/configs/batched/train.yaml:
    110. 

  110.           - ogbn-products
    111. 

  111.       - data/batched/hps/ogbn-products/best.json:
    112. 

  112.           - SAME
    113. 

  113.     deps:
    114. 

  114.       - experiments/scripts/batched/train.py
    115. 

  115.       - data/datasets/ogbn-products/
    116. 

  116.     outs:
    117. 

  117.       - data/batched/train/ogbn-products/
    118. 

  118. 

  119.   ### Additional experiments
    120. 

  120.   compare_augmentation_hyperparameter_sets:
    121. 

  121.     wdir: .
    122. 

  122.     cmd: PYTHONPATH=. python3 experiments/scripts/compare_augmentation_hyperparameter_sets.py
    123. 

  123.     deps:
    124. 

  124.       - experiments/scripts/compare_augmentation_hyperparameter_sets.py
    125. 

  125.       - data/full_batch/hps/
    126. 

  126.     outs:
    127. 

  127.       - data/augmentation_hyperparameter_plot.png
    128. 

  128. 

  129.   compare_running_times:
    130. 

  130.     wdir: .
    131. 

  131.     cmd: PYTHONPATH=. python3 experiments/scripts/time/compare_running_times.py
    132. 

  132.     deps:
    133. 

  133.       - experiments/scripts/time/compare_running_times.py
    134. 

  134.     outs:
    135. 

  135.       - data/time/log.csv
    136. 

  136.       - data/time/summary_table.txt
    137. 

  137. 

  138.   train_bgrl_full_batch:
    139. 

  139.     foreach:
    140. 

  140.       - WikiCS
    141. 

  141.       - Amazon-CS
    142. 

  142.       - Amazon-Photo
    143. 

  143.       - Coauthor-CS
    144. 

  144.       # - Coauthor-Physics  # OOM
    145. 

  145.       # - ogbn-arxiv
    146. 

  146.       # - PPI
    147. 

  147.     do:
    148. 

  148.       wdir: .
    149. 

  149.       cmd: PYTHONPATH=. python3 experiments/scripts/full_batch/train_bgrl.py ${item}
    150. 

  150.       params:
    151. 

  151.         - experiments/configs/full_batch/train_bgrl.yaml:
    152. 

  152.             - ${item}
    153. 

  153.       deps:
    154. 

  154.         - experiments/scripts/full_batch/train_bgrl.py
    155. 

  155.         - data/datasets/${item}/
    156. 

  156.       outs:
    157. 

  157.         - data/full_batch/train_bgrl/${item}/
    158. 

  158. 

  159.   bgrl_hps_batched:
    160. 

  160.     foreach:
    161. 

  161.       - ogbn-products
    162. 

  162.     do:
    163. 

  163.       wdir: .
    164. 

  164.       cmd: PYTHONPATH=. python3 experiments/scripts/batched/hps_bgrl.py ${item}
    165. 

  165.       params:
    166. 

  166.         - experiments/configs/batched/hps_bgrl.yaml:
    167. 

  167.             - ${item}
    168. 

  168.       deps:
    169. 

  169.         - experiments/scripts/batched/hps_bgrl.py
    170. 

  170.         - data/datasets/${item}/
    171. 

  171.       outs:
    172. 

  172.         - data/batched/hps_bgrl/${item}/log.csv
    173. 

  173.         - data/batched/hps_bgrl/${item}/best.json
    174. 

  174. 

  175.   bgrl_batched_train:
    176. 

  176.     foreach:
    177. 

  177.       - ogbn-products
    178. 

  178.     do:
    179. 

  179.       wdir: .
    180. 

  180.       cmd: PYTHONPATH=. python3 experiments/scripts/batched/train_bgrl.py ${item}
    181. 

  181.       params:
    182. 

  182.         - experiments/configs/batched/train_bgrl.yaml:
    183. 

  183.             - ${item}
    184. 

  184.         - data/batched/hps_bgrl/${item}/best.json:
    185. 

  185.             - SAME
    186. 

  186.       deps:
    187. 

  187.         - experiments/scripts/batched/train_bgrl.py
    188. 

  188.         - data/datasets/${item}/
    189. 

  189.       outs:
    190. 

  190.         - data/batched/train_bgrl/${item}/
    191. 

  191. 

  192.   evaluate_features_products:
    193. 

  193.     wdir: .
    194. 

  194.     cmd: PYTHONPATH=. python3 experiments/scripts/products/features.py
    195. 

  195.     deps:
    196. 

  196.       - experiments/scripts/products/features.py
    197. 

  197.       - data/datasets/ogbn-products/
    198. 

  198.     outs:
    199. 

  199.       - data/products/features.json
    200. 

  200. 

  201.   evaluate_deepwalk_products:
    202. 

  202.     wdir: .
    203. 

  203.     cmd: PYTHONPATH=. python3 experiments/scripts/products/deepwalk.py
    204. 

  204.     deps:
    205. 

  205.       - experiments/scripts/products/deepwalk.py
    206. 

  206.       - data/datasets/ogbn-products/
    207. 

  207.     outs:
    208. 

  208.       - data/products/deepwalk.json
    209. 

  209.       - data/products/deepwalk_features.json
    210.
Prev
Next
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Write Preview
Loading...
Integrate Google Cloud Storage
Use Google Storage
Select bucket
Upload key
Finish

Use Google Cloud Storage!

Browsing data directories saved to Google Cloud Storage is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your Google Storage bucket

Service Account Key

Congratulations!

graph-barlow-twins is now integrated with Google Cloud Storage!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes
Integrate AWS S3
Use S3 remote
Select bucket
Access key
Finish

Use AWS S3 as storage!

Browsing data directories saved to S3 is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your S3 bucket

Access key (If needed)

Congratulations!

graph-barlow-twins is now integrated with AWS S3!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes
Integrate S3 compatible storage
Use S3 like remote
Select bucket
Access key
Finish

Use any S3 compatible storage!

Browsing data directories saved to S3 compatible storage is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your S3 bucket

Access key (If needed)

Congratulations!

graph-barlow-twins is now integrated with your S3 compatible storage!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes
Integrate Azure Cloud Storage
Use Azure Storage
Select bucket
Set key
Finish

Use Azure Cloud Storage!

Browsing data directories saved to Azure Cloud Storage is possible with DAGsHub. Let's configure your repository to easily display your data in the context of any commit!

Specify your Azure Storage bucket

Access key (If needed)

Congratulations!

graph-barlow-twins is now integrated with Azure Cloud Storage!

Delete Storage Key

Are you sure you want to delete this access key?

No
Yes