From 3a6924cb8ff082eed0243b0684de88a858beee7d Mon Sep 17 00:00:00 2001
From: Shivaram Venkataraman <shivaram@eecs.berkeley.edu>
Date: Thu, 13 Jun 2013 13:25:10 -0700
Subject: [PATCH] Clean up some comments.

---
 ml/src/main/scala/spark/ml/GradientDescent.scala    | 1 +
 ml/src/main/scala/spark/ml/LogisticRegression.scala | 4 +---
 2 files changed, 2 insertions(+), 3 deletions(-)

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]) {