update

apache · Aug 17, 2017 · 7704930 · 7704930
1 parent eedc647
commit 7704930
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/ann/Layer.scala b/mllib/src/main/scala/org/apache/spark/ml/ann/Layer.scala
@@ -383,6 +383,8 @@ private[ann] trait TopologyModel extends Serializable {
    *
    * @param data input data
    * @return raw prediction
+   *
+   * Note: This interface is only used for classification Model.
    */
   def predictRaw(data: Vector): Vector
 
@@ -391,6 +393,8 @@ private[ann] trait TopologyModel extends Serializable {
    *
    * @param rawPrediction raw prediction vector
    * @return probability
+   *
+   * Note: This interface is only used for classification Model.
    */
   def raw2ProbabilityInPlace(rawPrediction: Vector): Vector