diff options
Diffstat (limited to 'python/demo_feast_wrapper.py')
| -rw-r--r-- | python/demo_feast_wrapper.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/python/demo_feast_wrapper.py b/python/demo_feast_wrapper.py index 0913788..6642ca8 100644 --- a/python/demo_feast_wrapper.py +++ b/python/demo_feast_wrapper.py @@ -1,5 +1,5 @@ #!/usr/bin/env python - +import feast import numpy as np ################################################################## @@ -34,7 +34,7 @@ data, labels = read_digits('digit.txt') n_observations = len(data) # number of samples in the data set n_features = len(data.transpose()) # number of features in the data set n_select = 15 # how many features to select -method = 'jmi' # feature selection algorithm +method = 'JMI' # feature selection algorithm print '---> Information' @@ -43,7 +43,5 @@ print ' :n_features - ' + str(n_features) print ' :n_select - ' + str(n_select) print ' :algorithm - ' + str(method) +selected_features = feast.select(data, labels, n_observations, n_features, n_select, method) -# Calvin: feature selection wrapper goes here -#selected_features = feast(data, labels, n_observations, n_features, n_select, \ -# method)
\ No newline at end of file |