Skip to content

Commit

Permalink
make IR experimental and update its doc
Browse files Browse the repository at this point in the history
  • Loading branch information
mengxr committed Feb 5, 2015
1 parent 9b8eed3 commit 55900f5
Showing 1 changed file with 11 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,13 @@ import java.util.Arrays.binarySearch

import scala.collection.mutable.ArrayBuffer

import org.apache.spark.annotation.Experimental
import org.apache.spark.api.java.{JavaDoubleRDD, JavaRDD}
import org.apache.spark.rdd.RDD

/**
* :: Experimental ::
*
* Regression model for isotonic regression.
*
* @param boundaries Array of boundaries for which predictions are known.
Expand All @@ -35,6 +38,7 @@ import org.apache.spark.rdd.RDD
* Results of isotonic regression and therefore monotone.
* @param isotonic indicates whether this is isotonic or antitonic.
*/
@Experimental
class IsotonicRegressionModel (
val boundaries: Array[Double],
val predictions: Array[Double],
Expand Down Expand Up @@ -123,21 +127,26 @@ class IsotonicRegressionModel (
}

/**
* :: Experimental ::
*
* Isotonic regression.
* Currently implemented using parallelized pool adjacent violators algorithm.
* Only univariate (single feature) algorithm supported.
*
* Sequential PAV implementation based on:
* Tibshirani, Ryan J., Holger Hoefling, and Robert Tibshirani.
* "Nearly-isotonic regression." Technometrics 53.1 (2011): 54-61.
* Available from http://www.stat.cmu.edu/~ryantibs/papers/neariso.pdf
* Available from [[http://www.stat.cmu.edu/~ryantibs/papers/neariso.pdf]]
*
* Sequential PAV parallelization based on:
* Kearsley, Anthony J., Richard A. Tapia, and Michael W. Trosset.
* "An approach to parallelizing isotonic regression."
* Applied Mathematics and Parallel Computing. Physica-Verlag HD, 1996. 141-147.
* Available from http://softlib.rice.edu/pub/CRPC-TRs/reports/CRPC-TR96640.pdf
* Available from [[http://softlib.rice.edu/pub/CRPC-TRs/reports/CRPC-TR96640.pdf]]
*
* @see [[http://en.wikipedia.org/wiki/Isotonic_regression Isotonic regression (Wikipedia)]]
*/
@Experimental
class IsotonicRegression private (private var isotonic: Boolean) extends Serializable {

/**
Expand Down

0 comments on commit 55900f5

Please sign in to comment.