Changeset 184 for trunk/EventBenchConsole/src/de/ugoe/cs/eventbench/commands/CMDtrainMarkovModel.java

Timestamp:

09/14/11 20:17:10 (13 years ago)

Author:

sherbold

Message:

• refactored commands for the training of models (trainDFA, trainMarkovModel, trainPPM) by extracting one common abstract superclass

+ added command updateModel for model reinforcement

File:

• trunk/EventBenchConsole/src/de/ugoe/cs/eventbench/commands/CMDtrainMarkovModel.java (modified) (4 diffs)

trunk/EventBenchConsole/src/de/ugoe/cs/eventbench/commands/CMDtrainMarkovModel.java

@@ -1 +1 @@
 package de.ugoe.cs.eventbench.commands;
 
-import java.security.InvalidParameterException;
 import java.util.List;
 import java.util.Random;
 
-import de.ugoe.cs.eventbench.data.Event;
-import de.ugoe.cs.eventbench.data.GlobalDataContainer;
 import de.ugoe.cs.eventbench.models.FirstOrderMarkovModel;
 import de.ugoe.cs.eventbench.models.HighOrderMarkovModel;
-import de.ugoe.cs.util.console.Command;
+import de.ugoe.cs.eventbench.models.TrieBasedModel;
 import de.ugoe.cs.util.console.Console;
 
@@ -18 +15 @@
  * 
  * @author Steffen Herbold
- * @version 1.0
+ * @version 2.0
  */
-public class CMDtrainMarkovModel implements Command {
+public class CMDtrainMarkovModel extends AbstractTrainCommand {
+
+        /**
+         * <p>
+         * Order of the Markov model.
+         * </p>
+         */
+        int order;
 
         /*
@@ -29 +33 @@
         @Override
         public void help() {
-                Console.println("Usage: trainMarkovModel <modelName> {<order>}");
+                Console.println("Usage: trainMarkovModel <modelName> <sequencesName> {<order>}");
+        }
+
+        /**
+         * <p>
+         * Handles the parameter order.
+         * </p>
+         * 
+         * @see de.ugoe.cs.eventbench.commands.AbstractTrainCommand#handleOptionalParameters(java.util.List)
+         */
+        @Override
+        void handleAdditionalParameters(List<Object> parameters) throws Exception {
+                if (parameters.size() >= 3) {
+                        order = Integer.parseInt((String) parameters.get(2));
+                }
         }
 
@@ -35 +53 @@
          * (non-Javadoc)
          * 
-         * @see de.ugoe.cs.util.console.Command#run(java.util.List)
+         * @see de.ugoe.cs.eventbench.commands.AbstractTrainCommand#createModel()
          */
-        @SuppressWarnings("unchecked")
         @Override
-        public void run(List<Object> parameters) {
-                String modelname;
-                int order = 1;
-                try {
-                        modelname = (String) parameters.get(0);
-                        if (parameters.size() == 2) {
-                                order = Integer.parseInt((String) parameters.get(1));
-                        }
-                } catch (Exception e) {
-                        throw new InvalidParameterException();
-                }
-
-                List<List<Event<?>>> sequences = null;
-                Object dataObject = GlobalDataContainer.getInstance().getData(
-                                "sequences");
-
-                try {
-                        sequences = (List<List<Event<?>>>) dataObject;
-                        if (sequences.size() > 0) {
-                                if (sequences.get(0).get(0) instanceof Event) {
-                                        HighOrderMarkovModel model;
-                                        if (order == 1) {
-                                                model = new FirstOrderMarkovModel(new Random());
-                                        } else {
-                                                model = new HighOrderMarkovModel(order, new Random());
-                                        }
-                                        model.train(sequences);
-                                        if (GlobalDataContainer.getInstance().addData(modelname,
-                                                        model)) {
-                                                Console.traceln("Old data \"" + modelname
-                                                                + "\" overwritten");
-                                        }
-                                } else {
-                                        Console.traceln("Illegal use of \"sequences\" parameter in the GlobalDataContainer.");
-                                        Console.traceln("The parameter should always be of type List<List<Event>>!");
-                                }
-                        }
-                } catch (ClassCastException e) {
-                        Console.println("Sequences need to be loaded first using parseXML");
+        TrieBasedModel createModel() {
+                if (order == 1) {
+                        return new FirstOrderMarkovModel(new Random());
+                } else {
+                        return new HighOrderMarkovModel(order, new Random());
                 }
         }