File tree Expand file tree Collapse file tree 1 file changed +12
-7
lines changed
algo/src/main/java/org/neo4j/gds/embeddings/node2vec Expand file tree Collapse file tree 1 file changed +12
-7
lines changed Original file line number Diff line number Diff line change @@ -75,26 +75,31 @@ public boolean next(long[] buffer) {
75
75
return false ;
76
76
}
77
77
78
+ private boolean skipWalk (long [] walk ){
79
+ int filteredWalkLength = filter (walk );
80
+ boolean skipWalk = filteredWalkLength < 2 ;
81
+ if (skipWalk ){
82
+ progressTracker .logProgress (); //a walk has been sampled and ignored
83
+ }
84
+ return skipWalk ;
85
+ }
78
86
private boolean nextWalk () {
79
87
if (attemptedSamplingWalks ){ //this means a walk has been exhausted
80
88
progressTracker .logProgress ();
81
89
}
82
- attemptedSamplingWalks = true ; //this is because first time nextWalk() is called, it doesnt have any walk lol
90
+ attemptedSamplingWalks = true ; //this is because first time nextWalk() is called, it doesnt have any walk
83
91
84
92
if (!walks .hasNext ()) {
85
93
return false ;
86
94
}
87
95
long [] walk = walks .next ();
88
96
89
- int filteredWalkLength = filter (walk );
90
-
91
- while (filteredWalkLength < 2 && walks .hasNext ()) {
97
+ boolean walkSkipped ;
98
+ while ((walkSkipped = skipWalk (walk )) && walks .hasNext ()) {
92
99
walk = walks .next ();
93
- filteredWalkLength = filter (walk );
94
-
95
100
}
96
101
97
- if (filteredWalkLength >= 2 ) {
102
+ if (! walkSkipped ) {
98
103
this .currentWalk = walk ;
99
104
centerWordIndex = -1 ;
100
105
return nextCenterWord ();
You can’t perform that action at this time.
0 commit comments