source: trunk/EventBenchConsole/src/de/ugoe/cs/eventbench/commands/CMDtrainPPM.java @ 84

Last change on this file since 84 was 84, checked in by sherbold, 13 years ago

+ added command generateRandomSequences

  • changed command generateReplayfile such that it accepts an optional parameter with the name of the sequences that are to be written, to allow writing sequences generated with generateRandomSequences
  • minor output changes
File size: 1.7 KB
Line 
1package de.ugoe.cs.eventbench.commands;
2
3import java.security.InvalidParameterException;
4import java.util.List;
5import java.util.Random;
6
7import de.ugoe.cs.eventbench.data.Event;
8import de.ugoe.cs.eventbench.data.GlobalDataContainer;
9import de.ugoe.cs.eventbench.models.PredictionByPartialMatch;
10import de.ugoe.cs.util.console.Command;
11import de.ugoe.cs.util.console.Console;
12
13public class CMDtrainPPM implements Command {
14
15        @Override
16        public void help() {
17                Console.println("Usage: trainPPM <modelName> <order>");
18        }
19
20        @SuppressWarnings("unchecked")
21        @Override
22        public void run(List<Object> parameters) {
23                String modelname;
24                int order;
25                try {
26                        modelname = (String) parameters.get(0);
27                        order = Integer.parseInt((String) parameters.get(1));
28                } catch (Exception e) {
29                        throw new InvalidParameterException();
30                }
31               
32                List<List<Event<?>>> sequences = null;
33                Object dataObject = GlobalDataContainer.getInstance().getData("sequences");
34                       
35                try {
36                        sequences = (List<List<Event<?>>>) dataObject;
37                        if( sequences.size()>0 ) {
38                                if( sequences.get(0).get(0) instanceof Event ) {
39                                        PredictionByPartialMatch model = new PredictionByPartialMatch(order, new Random());
40                                        model.train(sequences);
41                                        if( GlobalDataContainer.getInstance().addData(modelname, model) ) {
42                                                Console.traceln("Old data \"" + modelname + "\" overwritten");
43                                        }                                       
44                                } else {
45                                        Console.traceln("Illegal use of \"sequences\" parameter in the GlobalDataContainer.");
46                                        Console.traceln("The parameter should always be of type List<List<Event>>!");
47                                }
48                        }
49                }
50                catch(ClassCastException e) {
51                        Console.println("Sequences need to be loaded first using parseXML");
52                }
53        }
54
55}
Note: See TracBrowser for help on using the repository browser.