diff --git a/ml/src/main/scala/spark/ml/GradientDescent.scala b/ml/src/main/scala/spark/ml/GradientDescent.scala
index a03285bd53..564f89e5ee 100644
--- a/ml/src/main/scala/spark/ml/GradientDescent.scala
+++ b/ml/src/main/scala/spark/ml/GradientDescent.scala
@@ -12,6 +12,7 @@ object GradientDescent {
 
   /**
    * Run gradient descent in parallel using mini batches.
+   * Based on Matlab code written by John Duchi.
    *
    * @param data - Input data for SGD. RDD of form (label, [feature values]).
    * @param gradient - Gradient object that will be used to compute the gradient.
diff --git a/ml/src/main/scala/spark/ml/LogisticRegression.scala b/ml/src/main/scala/spark/ml/LogisticRegression.scala
index 76c4fe54c8..c134f8d77d 100644
--- a/ml/src/main/scala/spark/ml/LogisticRegression.scala
+++ b/ml/src/main/scala/spark/ml/LogisticRegression.scala
@@ -6,6 +6,7 @@ import org.jblas.DoubleMatrix
 
 /**
  * Logistic Regression using Stochastic Gradient Descent.
+ * Based on Matlab code written by John Duchi.
  */
 class LogisticRegressionModel(
   val weights: DoubleMatrix,
@@ -80,9 +81,6 @@ class LogisticRegression(var stepSize: Double, var miniBatchFraction: Double, va
   }
 }
 
-/**
- * Helper classes to build a LogisticRegression object.
- */
 object LogisticRegression {
 
   def main(args: Array[String]) {