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

[srowen]

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);