Fixed implicit casts

Project: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/repo
Commit: 
http://git-wip-us.apache.org/repos/asf/incubator-hivemall/commit/9d9eac0e
Tree: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/tree/9d9eac0e
Diff: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/diff/9d9eac0e

Branch: refs/heads/v0.5.0
Commit: 9d9eac0ebab6f546953c33d91e54abe91f996c72
Parents: 90461cb
Author: Makoto Yui <[email protected]>
Authored: Tue Jan 23 18:39:47 2018 +0900
Committer: Makoto Yui <[email protected]>
Committed: Tue Jan 23 18:39:47 2018 +0900

----------------------------------------------------------------------
 .../xgboost/tools/XGBoostMulticlassPredictUDTF.java  | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-hivemall/blob/9d9eac0e/xgboost/src/main/java/hivemall/xgboost/tools/XGBoostMulticlassPredictUDTF.java
----------------------------------------------------------------------
diff --git 
a/xgboost/src/main/java/hivemall/xgboost/tools/XGBoostMulticlassPredictUDTF.java
 
b/xgboost/src/main/java/hivemall/xgboost/tools/XGBoostMulticlassPredictUDTF.java
index fd67c09..b80f95a 100644
--- 
a/xgboost/src/main/java/hivemall/xgboost/tools/XGBoostMulticlassPredictUDTF.java
+++ 
b/xgboost/src/main/java/hivemall/xgboost/tools/XGBoostMulticlassPredictUDTF.java
@@ -18,8 +18,6 @@
  */
 package hivemall.xgboost.tools;
 
-import hivemall.utils.lang.Preconditions;
-
 import java.util.ArrayList;
 import java.util.List;
 
@@ -32,10 +30,11 @@ import 
org.apache.hadoop.hive.serde2.objectinspector.ObjectInspectorFactory;
 import org.apache.hadoop.hive.serde2.objectinspector.StructObjectInspector;
 import 
org.apache.hadoop.hive.serde2.objectinspector.primitive.PrimitiveObjectInspectorFactory;
 
-@Description(
-        name = "xgboost_multiclass_predict",
+import hivemall.utils.lang.Preconditions;
+
+@Description(name = "xgboost_multiclass_predict",
         value = "_FUNC_(string rowid, string[] features, string model_id, 
array<byte> pred_model [, string options]) "
-                + "- Returns a prediction result as (string rowid, int label, 
float probability)")
+                + "- Returns a prediction result as (string rowid, string 
label, float probability)")
 public final class XGBoostMulticlassPredictUDTF extends 
hivemall.xgboost.XGBoostPredictUDTF {
 
     public XGBoostMulticlassPredictUDTF() {
@@ -65,14 +64,14 @@ public final class XGBoostMulticlassPredictUDTF extends 
hivemall.xgboost.XGBoost
         final Object[] forwardObj = new Object[3];
         for (int i = 0, size = testData.size(); i < size; i++) {
             final float[] predicted_i = predicted[i];
-            final String rowId = testData.get(i).getRowId();
+            String rowId = testData.get(i).getRowId();
             forwardObj[0] = rowId;
 
             assert (predicted_i.length > 1);
             for (int j = 0; j < predicted_i.length; j++) {
-                forwardObj[1] = j;
+                forwardObj[1] = String.valueOf(j);
                 float prob = predicted_i[j];
-                forwardObj[2] = prob;
+                forwardObj[2] = Float.valueOf(prob);
                 forward(forwardObj);
             }
         }

Reply via email to