Skip to content

Commit c2e9783

Browse files
committed
Update Struc2vec and Struc2vec.walks is not initialized
1 parent 85ccd85 commit c2e9783

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/main/java/com/gnn/util/GNNHelper.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
import java.util.List;
1313
import java.util.Map;
1414
import java.util.Stack;
15-
import java.util.concurrent.atomic.AtomicInteger;
1615
import javax.swing.JFrame;
1716
import jsat.SimpleDataSet;
1817
import jsat.classifiers.CategoricalData;
@@ -279,7 +278,7 @@ public static <K> void evaluateEmbeddings(Map<K, List<Double>> embeddings, Strin
279278
try {
280279
int xSize = embeddings.values().iterator().next().size();
281280
ArrayList<Attribute> attributes = new ArrayList<Attribute>();
282-
for (int i=0; i<xSize; i++) {
281+
for (int i = 0; i < xSize; i++) {
283282
attributes.add(new Attribute("coordinate_" + i));
284283
}
285284
List<String> classes = new ArrayList<>(labels.keySet());
@@ -292,7 +291,7 @@ public static <K> void evaluateEmbeddings(Map<K, List<Double>> embeddings, Strin
292291
double num[] = new double[instances.numAttributes()];
293292
Instance instance = new DenseInstance(1, num);
294293
vs.forEach(v -> {
295-
for (int j=0; j<xSize; j++) {
294+
for (int j = 0; j < xSize; j++) {
296295
num[j] = embeddings.get(v).get(j);
297296
}
298297
num[instances.classIndex()] = instances.attribute(instances.classIndex()).indexOfValue(label);

0 commit comments

Comments
 (0)