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); } }
