diff --git a/.travis.yml b/.travis.yml index d70b562..5036d41 100644 --- a/.travis.yml +++ b/.travis.yml @@ -36,6 +36,7 @@ before_script: script: - coverage${PYVERSION} run --source ffx tests/x_square_test.py + - coverage${PYVERSION} run --source ffx tests/nonlinear_verbose_test.py after_success: - coveralls diff --git a/tests/nonlinear_verbose_test.py b/tests/nonlinear_verbose_test.py new file mode 100644 index 0000000..463d615 --- /dev/null +++ b/tests/nonlinear_verbose_test.py @@ -0,0 +1,19 @@ +#!/usr/bin/env python + +import numpy as np +import ffx + +# This creates a dataset of 1 predictor +train_X = np.array([[-5,-4,-3,-2,-1, 0, 1, 2, 3,4,5,6,7,8,9]]).T +train_y = np.array([4,4,3,2,1, 0, 1, 2, 3,4,4,4,4,4,4]) +test_X = np.array([[10,11,12]]).T +test_y = np.array([4, 4,4,]) + +models = ffx.run(train_X, train_y, test_X, test_y, ["x"],verbose = True) + +print('True model: y = min(abs(x),4)') +print('Results:') +print('Num bases,Test error (%),Model\n') +for model in models: + print('%10s, %13s, %s\n' % + ('%d' % model.numBases(), '%.4f' % (model.test_nmse * 100.0), model))