wip

ldbc · Sep 23, 2024 · 3a18aa3 · 3a18aa3
1 parent 05aa422
commit 3a18aa3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/src/main/java/ldbc/finbench/datagen/generation/dictionary/CommonTextDictionary.java b/src/main/java/ldbc/finbench/datagen/generation/dictionary/CommonTextDictionary.java
@@ -38,9 +38,10 @@ public String getUniformDistRandomText(Random random) {
     }
 
     public String getUniformDistRandomTextForComments(Random random) {
-        // Randomly select two resources and concatenate them to flatten the dataset
-        long index1 = random.nextInt(resources.size());
-        long index2 = random.nextInt(resources.size());
-        return resources.get(index1) + " " + resources.get(index2);
+        StringBuilder text = new StringBuilder();
+        for (int i = 0; i < 4; i++) {
+            text.append(random.nextInt(resources.size())).append(" ");
+        }
+        return text.toString();
     }
 }
diff --git a/src/main/resources/scale_factors.xml b/src/main/resources/scale_factors.xml
@@ -72,15 +72,15 @@
     <scale_factor name="1">
         <property>
             <name>generator.numPersons</name>
-            <value>20000</value>
+            <value>10000</value>
         </property>
         <property>
             <name>generator.numCompanies</name>
-            <value>20000</value>
+            <value>10000</value>
         </property>
         <property>
             <name>generator.numMediums</name>
-            <value>40000</value>
+            <value>20000</value>
         </property>
         <property>
             <name>transfer.minNumDegree</name>