Changeset 184 for trunk/EventBenchConsole/src/de/ugoe/cs/eventbench/commands/CMDtrainDFA.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/CMDtrainDFA.java (modified) (4 diffs)

trunk/EventBenchConsole/src/de/ugoe/cs/eventbench/commands/CMDtrainDFA.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.DeterministicFiniteAutomaton;
-import de.ugoe.cs.util.console.Command;
+import de.ugoe.cs.eventbench.models.TrieBasedModel;
 import de.ugoe.cs.util.console.Console;
 
@@ -17 +14 @@
  * 
  * @author Steffen Herbold
- * @version 1.0
+ * @version 2.0
  */
-public class CMDtrainDFA implements Command {
+public class CMDtrainDFA extends AbstractTrainCommand {
 
         /*
@@ -28 +25 @@
         @Override
         public void help() {
-                Console.println("Usage: trainDFA <modelName>");
+                Console.println("Usage: trainDFA <modelName> <sequencesName>");
+        }
+
+        /**
+         * <p>
+         * No additional parameters.
+         * </p>
+         * 
+         * @see de.ugoe.cs.eventbench.commands.AbstractTrainCommand#handleAdditionalParameters(java.util.List)
+         */
+        @Override
+        void handleAdditionalParameters(List<Object> parameters) throws Exception {
+                // no additional parameters.
         }
 
@@ -34 +43 @@
          * (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;
-                try {
-                        modelname = (String) parameters.get(0);
-                } 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) {
-                                        DeterministicFiniteAutomaton model = new DeterministicFiniteAutomaton(
-                                                        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<? extends Event<?>>!");
-                                }
-                        }
-                } catch (ClassCastException e) {
-                        Console.println("Sequences need to be loaded first using parseXML");
-                }
+        TrieBasedModel createModel() {
+                return new DeterministicFiniteAutomaton(new Random());
         }