Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Merge pull request #429 from iglesias/qda
QDA for python_modular's testsuite
  • Loading branch information
Soeren Sonnenburg committed Apr 7, 2012
2 parents ca1f345 + 51bf1b4 commit 8b6902c
Showing 1 changed file with 29 additions and 0 deletions.
29 changes: 29 additions & 0 deletions examples/undocumented/python_modular/classifier_qda_modular.py
@@ -0,0 +1,29 @@
from tools.load import LoadMatrix
lm = LoadMatrix()

traindat = lm.load_numbers('../data/fm_train_real.dat')
testdat = lm.load_numbers('../data/fm_test_real.dat')
label_traindat = lm.load_labels('../data/label_train_multiclass.dat')

parameter_list = [[traindat, testdat, label_traindat, 1e-4, False], \
[traindat, testdat, label_traindat, 1e-4, True]]

def classifier_qda_modular (fm_train_real=traindat, fm_test_real=testdat, label_train_twoclass=label_traindat, tolerance=1e-4, store_covs=False):
from shogun.Features import RealFeatures, Labels
from shogun.Classifier import QDA

feats_train = RealFeatures(fm_train_real)
feats_test = RealFeatures(fm_test_real)

labels = Labels(label_train_twoclass)

qda = QDA(feats_train, labels, tolerance, store_covs)
qda.train()

qda.apply(feats_test).get_labels()
qda.set_features(feats_test)
return qda, qda.apply().get_labels()

if __name__=='__main__':
print('QDA')
classifier_qda_modular(*parameter_list[0])

0 comments on commit 8b6902c

Please sign in to comment.