add regressor interface

BigTong · Dec 30, 2013 · a1331bd · a1331bd
1 parent 2133a10
commit a1331bd
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/regressor.go b/regressor.go
@@ -0,0 +1,16 @@
+package hector
+
+type Regressor interface {
+
+    //Set training parameters from parameter map
+    Init(params map[string]string)
+
+    //Train model on a given dataset
+    Train(dataset * DataSet)
+
+    //Predict the output of an input sample
+    Predict(sample * Sample) float64
+
+    SaveModel(path string)
+    LoadModel(path string)
+}