diff options
Diffstat (limited to 'test/test.py')
| -rw-r--r-- | test/test.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/test/test.py b/test/test.py index b5e16d1..caa3874 100644 --- a/test/test.py +++ b/test/test.py @@ -66,7 +66,7 @@ elif data_source == 'digits': 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 = 'MIM' # feature selection algorithm print '---> Information' @@ -147,6 +147,16 @@ if check_result(sf, n_relevant) == True: else: print ' mRMR failed!' +################################################################# +################################################################# +print ' Running MIM...' +sf = MIM(data, labels, n_select) +if check_result(sf, n_relevant) == True: + print ' MIM passed!' +else: + print ' MIM failed!' + +p print '---> Done unit tests!' |