Skip to content

Commit 33220ee

Browse files
committed
Expand logging test
1 parent cedd259 commit 33220ee

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

algo/src/test/java/org/neo4j/gds/embeddings/graphsage/algo/GraphSageTrainAlgorithmFactoryTest.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -470,7 +470,6 @@ void testLogging() {
470470
.batchSize(5)
471471
.randomSeed(42L)
472472
.modelName("model")
473-
.activationFunction("RELU")
474473
.epochs(2)
475474
.maxIterations(2)
476475
.build();
@@ -499,21 +498,21 @@ void testLogging() {
499498
"GraphSageTrain :: Train model :: Start",
500499
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Start",
501500
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 1 of 2 :: Start",
502-
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 1 of 2 :: LOSS: 132.63",
501+
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 1 of 2 :: LOSS: 132.52",
503502
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 1 of 2 100%",
504503
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 1 of 2 :: Finished",
505504
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 2 of 2 :: Start",
506-
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 2 of 2 :: LOSS: 129.13",
505+
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 2 of 2 :: LOSS: 130.73",
507506
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 2 of 2 100%",
508507
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Iteration 2 of 2 :: Finished",
509508
"GraphSageTrain :: Train model :: Epoch 1 of 2 :: Finished",
510509
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Start",
511510
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 1 of 2 :: Start",
512-
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 1 of 2 :: LOSS: 123.38",
511+
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 1 of 2 :: LOSS: 122.98",
513512
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 1 of 2 100%",
514513
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 1 of 2 :: Finished",
515514
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 2 of 2 :: Start",
516-
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 2 of 2 :: LOSS: 116.06",
515+
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 2 of 2 :: LOSS: 126.44",
517516
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 2 of 2 100%",
518517
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Iteration 2 of 2 :: Finished",
519518
"GraphSageTrain :: Train model :: Epoch 2 of 2 :: Finished",

0 commit comments

Comments
 (0)