71
71
@ ExtendWith (SoftAssertionsExtension .class )
72
72
class Node2VecTest extends BaseTest {
73
73
74
+ private static final List <Long > NO_SOURCE_NODES = List .of ();
75
+ private static final Optional <Long > NO_RANDOM_SEED = Optional .empty ();
76
+
74
77
private static final String DB_CYPHER =
75
78
"CREATE" +
76
79
" (a:Node1)" +
@@ -109,9 +112,12 @@ void embeddingsShouldHaveTheConfiguredDimension(String msg, Iterable<String> nod
109
112
embeddingDimension ,
110
113
EmbeddingInitializer .NORMALIZED
111
114
);
112
- HugeObjectArray <FloatVector > node2Vec = Node2Vec . create (
115
+ HugeObjectArray <FloatVector > node2Vec = new Node2Vec (
113
116
graph ,
114
117
4 ,
118
+ NO_SOURCE_NODES ,
119
+ NO_RANDOM_SEED ,
120
+ 1000 ,
115
121
new WalkParameters (10 , 80 , 1.0 , 1.0 , 0.001 , 0.75 ),
116
122
trainParameters ,
117
123
ProgressTracker .NULL_TRACKER
@@ -156,10 +162,12 @@ void shouldLogProgress(boolean relationshipWeights, int expectedProgresses) {
156
162
);
157
163
var log = Neo4jProxy .testLog ();
158
164
var progressTracker = new TestProgressTracker (progressTask , log , 4 , EmptyTaskRegistryFactory .INSTANCE );
159
- Node2Vec . create (
165
+ new Node2Vec (
160
166
graph ,
161
167
4 ,
162
- Optional .empty (),
168
+ NO_SOURCE_NODES ,
169
+ NO_RANDOM_SEED ,
170
+ 1000 ,
163
171
walkParameters ,
164
172
trainParameters ,
165
173
progressTracker
@@ -223,9 +231,12 @@ void failOnNegativeWeights() {
223
231
var walkParameters = new WalkParameters (10 , 80 , 1.0 , 1.0 , 0.001 , 0.75 );
224
232
var trainParameters = new TrainParameters (0.025 , 0.0001 , 1 , 1 , 1 , 128 , EmbeddingInitializer .NORMALIZED );
225
233
226
- var node2Vec = Node2Vec . create (
234
+ var node2Vec = new Node2Vec (
227
235
graph ,
228
236
4 ,
237
+ NO_SOURCE_NODES ,
238
+ NO_RANDOM_SEED ,
239
+ 1000 ,
229
240
walkParameters ,
230
241
trainParameters ,
231
242
ProgressTracker .NULL_TRACKER
@@ -248,19 +259,23 @@ void randomSeed(SoftAssertions softly) {
248
259
var walkParameters = new WalkParameters (1 , 20 , 1.0 , 1.0 , 0.001 , 0.75 );
249
260
var trainParameters = new TrainParameters (0.025 , 0.0001 , 1 , 1 , 1 , embeddingDimension , EmbeddingInitializer .NORMALIZED );
250
261
251
- var embeddings = Node2Vec . create (
262
+ var embeddings = new Node2Vec (
252
263
graph ,
253
264
4 ,
265
+ NO_SOURCE_NODES ,
254
266
Optional .of (1337L ),
267
+ 1000 ,
255
268
walkParameters ,
256
269
trainParameters ,
257
270
ProgressTracker .NULL_TRACKER
258
271
).compute ().embeddings ();
259
272
260
- var otherEmbeddings = Node2Vec . create (
273
+ var otherEmbeddings = new Node2Vec (
261
274
graph ,
262
275
4 ,
276
+ NO_SOURCE_NODES ,
263
277
Optional .of (1337L ),
278
+ 1000 ,
264
279
walkParameters ,
265
280
trainParameters ,
266
281
ProgressTracker .NULL_TRACKER
@@ -348,19 +363,23 @@ void shouldBeFairlyConsistentUnderOriginalIds(EmbeddingInitializer embeddingInit
348
363
var walkParameters = new WalkParameters (10 , 80 , 1.0 , 1.0 , 0.01 , 0.75 );
349
364
var trainParameters = new TrainParameters (0.025 , 0.0001 , 1 , 10 , 5 , embeddingDimension , embeddingInitializer );
350
365
351
- var firstEmbeddings = Node2Vec . create (
366
+ var firstEmbeddings = new Node2Vec (
352
367
firstGraph ,
353
368
4 ,
369
+ NO_SOURCE_NODES ,
354
370
Optional .of (1337L ),
371
+ 1000 ,
355
372
walkParameters ,
356
373
trainParameters ,
357
374
ProgressTracker .NULL_TRACKER
358
375
).compute ().embeddings ();
359
376
360
- var secondEmbeddings = Node2Vec . create (
377
+ var secondEmbeddings = new Node2Vec (
361
378
secondGraph ,
362
379
4 ,
380
+ NO_SOURCE_NODES ,
363
381
Optional .of (1337L ),
382
+ 1000 ,
364
383
walkParameters ,
365
384
trainParameters ,
366
385
ProgressTracker .NULL_TRACKER
0 commit comments