Changeset 9 for trunk/EventBenchCore/src/de/ugoe/cs/eventbench/ppm/PredictionByPartialMatch.java

Timestamp:

04/13/11 14:13:45 (13 years ago)

Author:

sherbold

Message:

• moved training logic for single sequences from PPM to Trie

File:

• trunk/EventBenchCore/src/de/ugoe/cs/eventbench/ppm/PredictionByPartialMatch.java (modified) (6 diffs)

trunk/EventBenchCore/src/de/ugoe/cs/eventbench/ppm/PredictionByPartialMatch.java

@@ -1 +1 @@
 package de.ugoe.cs.eventbench.ppm;
 
-import java.util.LinkedHashSet;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Random;
-import java.util.Set;
 
 import de.ugoe.cs.eventbench.data.Event;
@@ -12 +10 @@
 public class PredictionByPartialMatch {
         
-        private int maxOrder = 3;
+        private int maxOrder;
         
         private Trie<Event<?>> trie;
         
-        private Set<Event<?>> knownSymbols;
-        
         private double probEscape;
         
         private final Random r;
         
-        public PredictionByPartialMatch(Random r) {
-                this(r, 0.1);
-        }
-        
-        public PredictionByPartialMatch(Random r, double probEscape) {
+        public PredictionByPartialMatch(int maxOrder, Random r) {
+                this(maxOrder, r, 0.1);
+        }
+        
+        public PredictionByPartialMatch(int maxOrder, Random r, double probEscape) {
+                this.maxOrder = maxOrder;
                 this.r = r; // TODO defensive copy instead?
                 this.probEscape = probEscape;
@@ -42 +39 @@
         public void train(List<List<Event<?>>> sequences) {
                 trie = new Trie<Event<?>>();
-                knownSymbols = new LinkedHashSet<Event<?>>();
-                knownSymbols.add(Event.STARTEVENT);
-                knownSymbols.add(Event.ENDEVENT);
                 
                 for(List<Event<?>> sequence : sequences) {
@@ -51 +45 @@
                         currentSequence.add(Event.ENDEVENT);
                         
-                        addToTrie(currentSequence);
-                }
-        }
-        
-        private void addToTrie(List<Event<?>> sequence) {
+                        trie.train(currentSequence, maxOrder);
+                }
+        }
+        
+        /*private void addToTrie(List<Event<?>> sequence) {
                 if( knownSymbols==null ) {
                         knownSymbols = new LinkedHashSet<Event<?>>();
@@ -73 +67 @@
                         trie.add(sequence.subList(sequenceLength-j, sequenceLength));
                 }
-        }
+        }*/
         
         public List<? extends Event<?>> randomSequence() {
@@ -89 +83 @@
                         double probSum = 0.0;
                         List<Event<?>> currentContext = context.getLast(maxOrder);
-                        for( Event<?> symbol : knownSymbols ) {
+                        for( Event<?> symbol : trie.getKnownSymbols() ) {
                                 probSum += getProbability(currentContext, symbol);
                                 if( probSum>=randVal ) {