You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@mahout.apache.org by sr...@apache.org on 2011/04/18 23:32:31 UTC

svn commit: r1094764 - /mahout/trunk/examples/src/main/java/org/apache/mahout/classifier/sgd/TrainLogistic.java

Author: srowen
Date: Mon Apr 18 21:32:30 2011
New Revision: 1094764

URL: http://svn.apache.org/viewvc?rev=1094764&view=rev
Log:
MAHOUT-674 fix NPE by using right map key

Modified:
    mahout/trunk/examples/src/main/java/org/apache/mahout/classifier/sgd/TrainLogistic.java

Modified: mahout/trunk/examples/src/main/java/org/apache/mahout/classifier/sgd/TrainLogistic.java
URL: http://svn.apache.org/viewvc/mahout/trunk/examples/src/main/java/org/apache/mahout/classifier/sgd/TrainLogistic.java?rev=1094764&r1=1094763&r2=1094764&view=diff
==============================================================================
--- mahout/trunk/examples/src/main/java/org/apache/mahout/classifier/sgd/TrainLogistic.java (original)
+++ mahout/trunk/examples/src/main/java/org/apache/mahout/classifier/sgd/TrainLogistic.java Mon Apr 18 21:32:30 2011
@@ -114,7 +114,7 @@ public final class TrainLogistic {
       output.printf(Locale.ENGLISH, "%d\n", lmp.getNumFeatures());
       output.printf(Locale.ENGLISH, "%s ~ ", lmp.getTargetVariable());
       String sep = "";
-      for (String v : csv.getPredictors()) {
+      for (String v : csv.getTraceDictionary().keySet()) {
         double weight = predictorWeight(lr, 0, csv, v);
         if (weight != 0) {
           output.printf(Locale.ENGLISH, "%s%.3f*%s", sep, weight, v);