svn commit: r816557 - /lucene/mahout/trunk/core/src/main/java/org/apache/mahout/clustering/dirichlet/DirichletDriver.java

[is...@apache.org]

Author: isabel
Date: Fri Sep 18 09:24:23 2009
New Revision: 816557

URL: http://svn.apache.org/viewvc?rev=816557&view=rev
Log:
MAHOUT-138 - converted DirichletDriver

Modified:
    lucene/mahout/trunk/core/src/main/java/org/apache/mahout/clustering/dirichlet/DirichletDriver.java

Modified: lucene/mahout/trunk/core/src/main/java/org/apache/mahout/clustering/dirichlet/DirichletDriver.java
URL: http://svn.apache.org/viewvc/lucene/mahout/trunk/core/src/main/java/org/apache/mahout/clustering/dirichlet/DirichletDriver.java?rev=816557&r1=816556&r2=816557&view=diff
==============================================================================
--- lucene/mahout/trunk/core/src/main/java/org/apache/mahout/clustering/dirichlet/DirichletDriver.java (original)
+++ lucene/mahout/trunk/core/src/main/java/org/apache/mahout/clustering/dirichlet/DirichletDriver.java Fri Sep 18 09:24:23 2009
@@ -17,6 +17,14 @@
 
 package org.apache.mahout.clustering.dirichlet;
 
+import org.apache.commons.cli2.CommandLine;
+import org.apache.commons.cli2.Group;
+import org.apache.commons.cli2.Option;
+import org.apache.commons.cli2.OptionException;
+import org.apache.commons.cli2.builder.ArgumentBuilder;
+import org.apache.commons.cli2.builder.DefaultOptionBuilder;
+import org.apache.commons.cli2.builder.GroupBuilder;
+import org.apache.commons.cli2.commandline.Parser;
 import org.apache.hadoop.fs.FileSystem;
 import org.apache.hadoop.fs.Path;
 import org.apache.hadoop.io.SequenceFile;
@@ -31,6 +39,7 @@
 import org.apache.mahout.clustering.kmeans.KMeansDriver;
 import org.apache.mahout.matrix.SparseVector;
 import org.apache.mahout.matrix.Vector;
+import org.apache.mahout.utils.CommandLineUtil;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -54,15 +63,66 @@
 
   public static void main(String[] args) throws InstantiationException,
       IllegalAccessException, ClassNotFoundException, IOException {
-    String input = args[0];
-    String output = args[1];
-    String modelFactory = args[2];
-    int numClusters = Integer.parseInt(args[3]);
-    int maxIterations = Integer.parseInt(args[4]);
-    double alpha_0 = Double.parseDouble(args[5]);
-    int numReducers = Integer.parseInt(args[6]);
-    runJob(input, output, modelFactory, numClusters, maxIterations, alpha_0,
-        numReducers);
+    DefaultOptionBuilder obuilder = new DefaultOptionBuilder();
+    ArgumentBuilder abuilder = new ArgumentBuilder();
+    GroupBuilder gbuilder = new GroupBuilder();
+
+    Option inputOpt = obuilder.withLongName("input").withRequired(true).withShortName("i").
+        withArgument(abuilder.withName("input").withMinimum(1).withMaximum(1).create()).
+        withDescription("The Path for input Vectors. Must be a SequenceFile of Writable, Vector").withShortName("i").create();
+    
+    Option outputOpt = obuilder.withLongName("output").withRequired(true).withShortName("o").
+        withArgument(abuilder.withName("output").withMinimum(1).withMaximum(1).create()).
+        withDescription("The directory pathname for output points.").create();
+    
+    Option maxIterOpt = obuilder.withLongName("maxIter").withRequired(true).withShortName("x").
+        withArgument(abuilder.withName("maxIter").withMinimum(1).withMaximum(1).create()).
+        withDescription("The maximum number of iterations.").create();
+    
+    Option topicsOpt = obuilder.withLongName("numModels").withRequired(true).withArgument(
+        abuilder.withName("numModels").withMinimum(1).withMaximum(1).create()).withDescription(
+        "The number of models").withShortName("k").create();
+
+    Option mOpt = obuilder.withLongName("alpha").withRequired(true).withShortName("m").
+        withArgument(abuilder.withName("alpha").withMinimum(1).withMaximum(1).create()).
+        withDescription("The alpha0 value for the DirichletDistribution.").create();
+
+    Option modelOpt = obuilder.withLongName("modelClass").withRequired(true).withShortName("d").
+        withArgument(abuilder.withName("modelClass").withMinimum(1).withMaximum(1).create()).
+        withDescription("The ModelDistribution class name.").create();
+
+    Option numRedOpt = obuilder.withLongName("maxRed").withRequired(true).withShortName("r").
+        withArgument(abuilder.withName("maxRed").withMinimum(1).withMaximum(1).create()).
+        withDescription("The number of reduce tasks.").create();
+
+    Option helpOpt = obuilder.withLongName("help").
+        withDescription("Print out help").withShortName("h").create();
+
+    Group group = gbuilder.withName("Options").withOption(inputOpt).withOption(outputOpt).withOption(modelOpt).
+        withOption(maxIterOpt).withOption(mOpt).withOption(topicsOpt).withOption(helpOpt).
+        withOption(numRedOpt).create();
+
+    try {
+      Parser parser = new Parser();
+      parser.setGroup(group);
+      CommandLine cmdLine = parser.parse(args);
+      if (cmdLine.hasOption(helpOpt)) {
+        CommandLineUtil.printHelp(group);
+        return;
+      }
+
+      String input = cmdLine.getValue(inputOpt).toString();
+      String output = cmdLine.getValue(outputOpt).toString();
+      String modelFactory = cmdLine.getValue(modelOpt).toString();
+      int numReducers = Integer.parseInt(cmdLine.getValue(numRedOpt).toString());
+      int numModels = Integer.parseInt(cmdLine.getValue(topicsOpt).toString());
+      int maxIterations = Integer.parseInt(cmdLine.getValue(maxIterOpt).toString());
+      double alpha_0 = Double.parseDouble(cmdLine.getValue(mOpt).toString());
+      runJob(input, output, modelFactory, numModels, maxIterations, alpha_0, numReducers);
+    } catch (OptionException e) {
+      log.error("Exception parsing command line: ", e);
+      CommandLineUtil.printHelp(group);
+    }
   }
 
   /**