Last change
on this file since 334 was
329,
checked in by sherbold, 13 years ago
|
|
-
Property svn:mime-type set to
text/plain
|
File size:
726 bytes
|
Rev | Line | |
---|
[326] | 1 | package de.ugoe.cs.eventbench.models;
|
---|
| 2 |
|
---|
| 3 | import java.util.ArrayList;
|
---|
| 4 | import java.util.List;
|
---|
| 5 | import java.util.Random;
|
---|
| 6 |
|
---|
| 7 | import de.ugoe.cs.eventbench.data.Event;
|
---|
| 8 |
|
---|
| 9 | public 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<?>>();
|
---|
[329] | 20 | if( context.isEmpty() ) {
|
---|
| 21 | return 2;
|
---|
| 22 | }
|
---|
[326] | 23 | list.add(context.get(context.size()-1));
|
---|
| 24 | if( trie.find(list).getFollowingSymbols().contains(symbol) ) {
|
---|
| 25 | return 1;
|
---|
| 26 | } else {
|
---|
| 27 | return 0;
|
---|
| 28 | }
|
---|
| 29 | }
|
---|
| 30 | } |
---|
Note: See
TracBrowser
for help on using the repository browser.