From ac8e950227b0c38a7e35dcc226873971a56a9924 Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Sat, 5 Mar 2016 12:12:43 -0800
Subject: [PATCH] Add Labeled Point, minor fix build

---
 jvm-packages/pom.xml                          |  1 +
 jvm-packages/xgboost4j-flink/pom.xml          | 14 +++---
 .../scala/ml/dmlc/xgboost4j/flink/Test.scala  |  3 +-
 jvm-packages/xgboost4j-spark/pom.xml          |  2 +-
 .../java/ml/dmlc/xgboost4j/LabeledPoint.java  | 45 +++++++++++++++++++
 5 files changed, 55 insertions(+), 10 deletions(-)
 create mode 100644 jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/LabeledPoint.java

diff --git a/jvm-packages/pom.xml b/jvm-packages/pom.xml
index ce9883059..dc11e307d 100644
--- a/jvm-packages/pom.xml
+++ b/jvm-packages/pom.xml
@@ -20,6 +20,7 @@
     <modules>
         <module>xgboost4j</module>
         <module>xgboost4j-demo</module>
+        <module>xgboost4j-flink</module>
         <module>xgboost4j-spark</module>
     </modules>
     <build>
diff --git a/jvm-packages/xgboost4j-flink/pom.xml b/jvm-packages/xgboost4j-flink/pom.xml
index 9c7b55427..ae49c2295 100644
--- a/jvm-packages/xgboost4j-flink/pom.xml
+++ b/jvm-packages/xgboost4j-flink/pom.xml
@@ -4,7 +4,7 @@
          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
     <modelVersion>4.0.0</modelVersion>
     <parent>
-        <groupId>org.dmlc</groupId>
+        <groupId>ml.dmlc</groupId>
         <artifactId>xgboostjvm</artifactId>
         <version>0.1</version>
     </parent>
@@ -13,7 +13,7 @@
     <packaging>jar</packaging>
     <dependencies>
         <dependency>
-            <groupId>org.dmlc</groupId>
+            <groupId>ml.dmlc</groupId>
             <artifactId>xgboost4j</artifactId>
             <version>0.1</version>
         </dependency>
@@ -24,24 +24,24 @@
         </dependency>
         <dependency>
             <groupId>org.apache.flink</groupId>
-            <artifactId>flink-java_2.11</artifactId>
+            <artifactId>flink-java</artifactId>
             <version>0.10.2</version>
         </dependency>
         <dependency>
             <groupId>org.apache.flink</groupId>
-            <artifactId>flink-scala_2.11</artifactId>
+            <artifactId>flink-scala</artifactId>
             <version>0.10.2</version>
         </dependency>
         <dependency>
             <groupId>org.apache.flink</groupId>
-            <artifactId>flink-clients_2.11</artifactId>
+            <artifactId>flink-clients</artifactId>
             <version>0.10.2</version>
         </dependency>
         <dependency>
             <groupId>org.apache.flink</groupId>
-            <artifactId>flink-ml_2.11</artifactId>
+            <artifactId>flink-ml</artifactId>
             <version>0.10.2</version>
         </dependency>
     </dependencies>
 
-</project>
\ No newline at end of file
+</project>
diff --git a/jvm-packages/xgboost4j-flink/src/main/scala/ml/dmlc/xgboost4j/flink/Test.scala b/jvm-packages/xgboost4j-flink/src/main/scala/ml/dmlc/xgboost4j/flink/Test.scala
index f4934fc2f..1beec66b3 100644
--- a/jvm-packages/xgboost4j-flink/src/main/scala/ml/dmlc/xgboost4j/flink/Test.scala
+++ b/jvm-packages/xgboost4j-flink/src/main/scala/ml/dmlc/xgboost4j/flink/Test.scala
@@ -16,8 +16,7 @@
 
 package ml.dmlc.xgboost4j.flink
 
-import ml.dmlc.xgboost4j.java.{Rabit,RabitTracker}
-
+import ml.dmlc.xgboost4j.java.{Rabit, RabitTracker}
 import ml.dmlc.xgboost4j.scala.Booster
 import ml.dmlc.xgboost4j.scala.DMatrix
 import ml.dmlc.xgboost4j.scala.XGBoost
diff --git a/jvm-packages/xgboost4j-spark/pom.xml b/jvm-packages/xgboost4j-spark/pom.xml
index cb7290654..4e40b83e4 100644
--- a/jvm-packages/xgboost4j-spark/pom.xml
+++ b/jvm-packages/xgboost4j-spark/pom.xml
@@ -8,7 +8,7 @@
         <artifactId>xgboostjvm</artifactId>
         <version>0.1</version>
     </parent>
-    <artifactId>xgboost4jspark</artifactId>
+    <artifactId>xgboost4j-spark</artifactId>
     <build>
         <plugins>
             <plugin>
diff --git a/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/LabeledPoint.java b/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/LabeledPoint.java
new file mode 100644
index 000000000..19d4b7da9
--- /dev/null
+++ b/jvm-packages/xgboost4j/src/main/java/ml/dmlc/xgboost4j/LabeledPoint.java
@@ -0,0 +1,45 @@
+package ml.dmlc.xgboost4j;
+
+/**
+ * Labeled data point for training examples.
+ * Represent a sparse training instance.
+ */
+public class LabeledPoint {
+  /** Label of the point */
+  float label;
+  /** Weight of this data point */
+  float weight = 1.0f;
+  /** Feature indices, used for sparse input */
+  int[] indices = null;
+  /** Feature values */
+  float[] values;
+
+  private LabeledPoint() {}
+
+  /**
+   * Create Labeled data point from sparse vector.
+   * @param label The label of the data point.
+   * @param indices The indices
+   * @param values The values.
+   */
+  public static LabeledPoint fromSparseVector(float label, int[] indices, float[] values) {
+    LabeledPoint ret = new LabeledPoint();
+    ret.label = label;
+    ret.indices = indices;
+    ret.values = values;
+    return ret;
+  }
+
+  /**
+   * Create Labeled data point from dense vector.
+   * @param label The label of the data point.
+   * @param values The values.
+   */
+  public static LabeledPoint fromDenseVector(float label, float[] values) {
+    LabeledPoint ret = new LabeledPoint();
+    ret.label = label;
+    ret.indices = null;
+    ret.values = values;
+    return ret;
+  }
+}