source: trunk/EventBenchCoreTest/src/de/ugoe/cs/eventbench/models/MockTrieBasedModel.java @ 326

Last change on this file since 326 was 326, checked in by sherbold, 12 years ago
  • changed signature of de.ugoe.cs.util.eventbench.models.TrieBasedModel?.train() and update() from using Collection<List<Event<?>>> to Collection<List<? extends Event<?>>>
  • Property svn:mime-type set to text/plain
File size: 678 bytes
Line 
1package de.ugoe.cs.eventbench.models;
2
3import java.util.ArrayList;
4import java.util.List;
5import java.util.Random;
6
7import de.ugoe.cs.eventbench.data.Event;
8
9public class MockTrieBasedModel extends TrieBasedModel {
10        private static final long serialVersionUID = 1L;
11
12        public MockTrieBasedModel(int markovOrder, Random r) {
13                super(markovOrder, r);
14        }
15
16        @Override
17        public double getProbability(List<? extends Event<?>> context,
18                        Event<?> symbol) {
19                List<Event<?>> list = new ArrayList<Event<?>>();
20                list.add(context.get(context.size()-1));
21                if( trie.find(list).getFollowingSymbols().contains(symbol) ) {
22                        return 1;
23                } else {
24                        return 0;
25                }
26        }
27}
Note: See TracBrowser for help on using the repository browser.