Skip to content

Commit 34d2558

Browse files
committed
Added the possibility to override the computation of weights on edges
1 parent eb5e913 commit 34d2558

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

src/main/java/ldbc/snb/datagen/generator/LDBCDatagen.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,12 @@ public static synchronized void init (Configuration conf) {
6262
DatagenParams.readConf(conf);
6363
Dictionaries.loadDictionaries(conf);
6464
SN.initialize();
65-
Person.personSimilarity = new GeoDistanceSimilarity();
65+
try {
66+
Person.personSimilarity = (Person.PersonSimilarity) Class.forName(conf.get("ldbc.snb.datagen.generator.person.similarity")).newInstance();
67+
} catch(Exception e) {
68+
System.err.println("Error while loading person similarity class");
69+
System.err.println(e.getMessage());
70+
}
6671
initialized = true;
6772
}
6873
}

src/main/java/ldbc/snb/datagen/util/ConfigParser.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ public static Configuration initialize() {
5050
conf.set("ldbc.snb.datagen.serializer.dateFormatter", "ldbc.snb.datagen.serializer.formatter.StringDateFormatter");
5151
conf.set("ldbc.snb.datagen.serializer.formatter.StringDateFormatter.dateTimeFormat", "yyyy-MM-dd'T'HH:mm:ss.SSSZ");
5252
conf.set("ldbc.snb.datagen.serializer.formatter.StringDateFormatter.dateFormat", "yyyy-MM-dd");
53+
conf.set("ldbc.snb.datagen.generator.person.similarity", "ldbc.snb.datagen.objects.similarity.GeoDistanceSimilarity");
5354

5455
/** Loading predefined Scale Factors **/
5556

0 commit comments

Comments
 (0)