Skip to content

Commit d065911

Browse files
committed
Log max likelihood obj in N2V
1 parent 65310b4 commit d065911

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

algo/src/main/java/org/neo4j/gds/embeddings/node2vec/Node2VecModel.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@
2020
package org.neo4j.gds.embeddings.node2vec;
2121

2222
import org.neo4j.gds.annotation.ValueClass;
23+
import org.neo4j.gds.collections.ha.HugeObjectArray;
2324
import org.neo4j.gds.core.concurrency.RunWithConcurrency;
2425
import org.neo4j.gds.core.utils.mem.MemoryEstimation;
2526
import org.neo4j.gds.core.utils.mem.MemoryEstimations;
26-
import org.neo4j.gds.collections.ha.HugeObjectArray;
2727
import org.neo4j.gds.core.utils.partition.PartitionUtils;
2828
import org.neo4j.gds.core.utils.progress.tasks.ProgressTracker;
2929
import org.neo4j.gds.mem.BitUtil;
@@ -184,7 +184,7 @@ Result train() {
184184
.run();
185185

186186
double loss = tasks.stream().mapToDouble(TrainingTask::lossSum).sum();
187-
progressTracker.logInfo(formatWithLocale("Loss %.4f", loss));
187+
progressTracker.logInfo(formatWithLocale("Maximum likelihood objective is %.4f", loss));
188188
lossPerIteration.add(loss);
189189

190190
progressTracker.endSubTask();

0 commit comments

Comments
 (0)