Fixes for changes in the type of seed

myui · Jun 7, 2016 · cf28b83 · cf28b83
1 parent a23062b
commit cf28b83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spark/src/main/java/hivemall/dataset/LogisticRegressionDataGeneratorUDTFWrapper.java b/spark/src/main/java/hivemall/dataset/LogisticRegressionDataGeneratorUDTFWrapper.java
@@ -83,7 +83,7 @@ public StructObjectInspector initialize(ObjectInspector[] argOIs) throws UDFArgu
             Field rnd2 = clazz.getDeclaredField("rnd2");
             Field r_seed = clazz.getDeclaredField("r_seed");
             r_seed.setAccessible(true);
-            final int seed = r_seed.getInt(udtf) + (int) Thread.currentThread().getId();
+            final long seed = r_seed.getLong(udtf) + (int) Thread.currentThread().getId();
             rnd1.setAccessible(true);
             rnd2.setAccessible(true);
             rnd1.set(udtf, new Random(seed));