Skip to content

Commit fa54488

Browse files
orazvebreakanalysis
andcommitted
Add learning rate
Co-authored-by: Jacob Sznajdman <[email protected]>
1 parent 74d5639 commit fa54488

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graphdatascience/gnn/gnn_nc_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
class GNNNodeClassificationRunner(UncallableNamespace, IllegalAttrChecker):
99
def make_graph_sage_config(self, graph_sage_config):
1010
GRAPH_SAGE_DEFAULT_CONFIG = {"layer_config": {}, "num_neighbors": [25, 10], "dropout": 0.5,
11-
"hidden_channels": 256}
11+
"hidden_channels": 256, "learning_rate": 0.003}
1212
final_sage_config = GRAPH_SAGE_DEFAULT_CONFIG
1313
if graph_sage_config:
1414
bad_keys = []

0 commit comments

Comments
 (0)