aboutsummaryrefslogtreecommitdiff
path: root/mllib
diff options
context:
space:
mode:
authorSean Owen <sowen@cloudera.com>2014-04-18 10:04:02 -0700
committerReynold Xin <rxin@apache.org>2014-04-18 10:04:02 -0700
commit8aa1f4c4f6d60168737699b5a9eafd6a05660976 (patch)
tree768546a0b82f9f76feb31d75c6037c962e5d1bab /mllib
parentaa17f022c59af02b04b977da9017671ef14d664a (diff)
downloadspark-8aa1f4c4f6d60168737699b5a9eafd6a05660976.tar.gz
spark-8aa1f4c4f6d60168737699b5a9eafd6a05660976.tar.bz2
spark-8aa1f4c4f6d60168737699b5a9eafd6a05660976.zip
SPARK-1357 (addendum). More Experimental items in MLlib
Per discussion, this is my suggestion to make ALS Rating, ClassificationModel, RegressionModel experimental for now, to reserve the right to possibly change after 1.0. See what you think of this much. Author: Sean Owen <sowen@cloudera.com> Closes #372 from srowen/SPARK-1357Addendum and squashes the following commits: 17cf1ea [Sean Owen] Remove (another) blank line after ":: Experimental ::" 6800e4c [Sean Owen] Remove blank line after ":: Experimental ::" b3a88d2 [Sean Owen] Make ALS Rating, ClassificationModel, RegressionModel experimental for now, to reserve the right to possibly change after 1.0
Diffstat (limited to 'mllib')
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/classification/ClassificationModel.scala3
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala2
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/regression/RegressionModel.scala5
3 files changed, 10 insertions, 0 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/classification/ClassificationModel.scala b/mllib/src/main/scala/org/apache/spark/mllib/classification/ClassificationModel.scala
index bd10e2e9e1..6332301e30 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/classification/ClassificationModel.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/classification/ClassificationModel.scala
@@ -19,11 +19,14 @@ package org.apache.spark.mllib.classification
import org.apache.spark.mllib.linalg.Vector
import org.apache.spark.rdd.RDD
+import org.apache.spark.annotation.Experimental
/**
+ * :: Experimental ::
* Represents a classification model that predicts to which of a set of categories an example
* belongs. The categories are represented by double values: 0.0, 1.0, 2.0, etc.
*/
+@Experimental
trait ClassificationModel extends Serializable {
/**
* Predict values for the given data set using the model trained.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala b/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala
index 5cc47de8ff..102742c7c5 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala
@@ -56,8 +56,10 @@ private[recommendation] case class InLinkBlock(
/**
+ * :: Experimental ::
* A more compact class to represent a rating than Tuple3[Int, Int, Double].
*/
+@Experimental
case class Rating(val user: Int, val product: Int, val rating: Double)
/**
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/regression/RegressionModel.scala b/mllib/src/main/scala/org/apache/spark/mllib/regression/RegressionModel.scala
index 5e4b8a345b..027305aa78 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/regression/RegressionModel.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/regression/RegressionModel.scala
@@ -19,7 +19,12 @@ package org.apache.spark.mllib.regression
import org.apache.spark.rdd.RDD
import org.apache.spark.mllib.linalg.Vector
+import org.apache.spark.annotation.Experimental
+/**
+ * :: Experimental ::
+ */
+@Experimental
trait RegressionModel extends Serializable {
/**
* Predict values for the given data set using the model trained.