Changeset 258 for trunk/EventBenchCore/src/de/ugoe/cs/eventbench/models/Trie.java

Timestamp:

10/15/11 01:41:55 (13 years ago)

Author:

sherbold

Message:

• added methods getChildren() and setCount() to de.ugoe.cs.eventbench.models.TrieNode?
• added method updateKnownSymbols() to de.ugoe.cs.eventbench.models.Trie
• changed the visibility of some members of de.ugoe.cs.eventbench.models.PredictionByPartionMatch?
• added class de.ugoe.cs.eventbench.models.ModelFlattener? to generate FirstOrderMarkovModel? from higher-order models through state splitting
• tweaked entropy calculation of de.ugoe.cs.eventbench.models.FirstOrderMarkovModel? to work with flattened models (locating of START and END was a problem)

File:

• trunk/EventBenchCore/src/de/ugoe/cs/eventbench/models/Trie.java (modified) (2 diffs)

trunk/EventBenchCore/src/de/ugoe/cs/eventbench/models/Trie.java

@@ -85 +85 @@
          */
         public void train(List<T> sequence, int maxOrder) {
-                if( maxOrder<1 ) {
+                if (maxOrder < 1) {
                         return;
                 }
@@ -400 +400 @@
                 return rootNode.getNumLeafs();
         }
+
+        /**
+         * <p>
+         * Updates the list of known symbols by replacing it with all symbols that
+         * are found in the child nodes of the root node. This should be the same as
+         * all symbols that are contained in the trie.
+         * </p>
+         */
+        public void updateKnownSymbols() {
+                knownSymbols = new HashSet<T>();
+                for (TrieNode<T> node : rootNode.getChildren()) {
+                        knownSymbols.add(node.getSymbol());
+                }
+        }
 }