Add data model for predictor

JIRA: PREDICTION-36

Change-Id: I871ce7e3696e7154ee4adb5c83cdb45c02a97d7b
Signed-off-by: Hai Liu <hai.liu@huawei.com>

1 files changed, 32 insertions, 0 deletions

diff --git a/src/model/ModelInterface.java b/src/model/ModelInterface.java
new file mode 100644
index 0000000..a4d1b7b
--- /dev/null
+++ b/src/model/ModelInterface.java
@@ -0,0 +1,32 @@
+package model;
+
+import java.util.ArrayList;
+
+import weka.core.Instances;
+
+import predictor.*;
+
+
+public interface ModelInterface {
+	public String getDatapath();
+	public ArrayList<PredictorInterface> getPredictors();
+	
+	public void loadTrainingData(String path);
+	public void loadRawLog(String path);
+	public Instances getTrainingInstances();
+	public void setPreprocessedInstances(Instances instances);
+	public Instances getPreprocessedInstances();
+	public void savePreprocessedInstances(String path);
+	public void addPredictor(String shortName);
+	public void crossValidatePredictors(int numFold);
+	public void crossValidatePredictors(int numFold, long seed);
+	public void selectTrainingMethod();
+	public void trainPredictors() throws Exception;
+	public void benchmark(int rounds, String filename) throws Exception;
+
+	public String getPredictorNames();
+	public String toString();
+	public void saveSettings(String filename) throws Exception;
+	public void saveResults(String filename) throws Exception;
+}
+