working on feast ctypes

1 files changed, 25 insertions, 0 deletions

diff --git a/python/feast.py b/python/feast.py
new file mode 100644
index 0000000..bc6fc93
--- /dev/null
+++ b/python/feast.py
@@ -0,0 +1,25 @@
+import numpy as np
+from ctypes import * 
+
+def select(data, labels, n_observations, n_features, n_select, method):
+  
+  selected_features = []
+
+  try:
+    libFSToolbox = CDLL("libFSToolbox.so"); 
+  except:
+    print "Error: could not find libFSToolbox"
+    exit()
+
+# JMI(n_features_to_ret, int n_samples, int n_feats, double *featureMatrix, double *classcol, outputFeatures);
+  c_output = (c_double * n_select)
+  c_n_observations = c_int(n_observations)
+  c_n_select = c_int(n_select)
+  c_n_features = c_int(n_features)
+  c_data = (c_double * n_observations * n_features)(*data.tolist())
+  c_labels = (c_int * len(labels))(*labels)
+
+  # right now just call only JMI, work out the rest later
+  libFSToolbox.JMI(c_n_select, c_n_observations, c_n_features, c_data, c_labels, c_output)
+
+  return selected_features