@@ -375,7 +375,7 @@ object FactorGenerationStage extends DatagenStage with Logging {
375
375
val allKnows = undirectedKnows(knows).cache()
376
376
val minSampleSize = 100.0
377
377
378
- val chinesePeopleSample = (relations : DataFrame ) => {
378
+ val peopleInChinaSample = (relations : DataFrame ) => {
379
379
val peopleInChina = person
380
380
.as(" Person" )
381
381
.join(cities.as(" City" ), $" City.id" === $" Person.LocationCityId" )
@@ -402,7 +402,7 @@ object FactorGenerationStage extends DatagenStage with Logging {
402
402
allKnows,
403
403
n = 4 ,
404
404
joinKeys = (" Person2Id" , " Person1Id" ),
405
- sample = Some (chinesePeopleSample )
405
+ sample = Some (peopleInChinaSample )
406
406
)
407
407
.join(person.as(" Person1" ), $" Person1.id" === $" Person1Id" )
408
408
.join(person.as(" Person2" ), $" Person2.id" === $" Person1Id" )
@@ -423,7 +423,7 @@ object FactorGenerationStage extends DatagenStage with Logging {
423
423
val allKnows = undirectedKnows(knows).cache()
424
424
val minSampleSize = 100.0
425
425
426
- val chinesePeopleSample = (relations : DataFrame ) => {
426
+ val peopleInChinaSample = (relations : DataFrame ) => {
427
427
val peopleInChina = person
428
428
.as(" Person" )
429
429
.join(cities.as(" City" ), $" City.id" === $" Person.LocationCityId" )
@@ -450,7 +450,7 @@ object FactorGenerationStage extends DatagenStage with Logging {
450
450
allKnows,
451
451
n = 2 ,
452
452
joinKeys = (" Person2Id" , " Person1Id" ),
453
- sample = Some (chinesePeopleSample )
453
+ sample = Some (peopleInChinaSample )
454
454
)
455
455
.join(person.as(" Person1" ), $" Person1.id" === $" Person1Id" )
456
456
.join(person.as(" Person2" ), $" Person2.id" === $" Person1Id" )
0 commit comments