Python GradientStepの例

コード例 #1

ファイル: test_quadratic.py プロジェクト: repagh/pydstool

def test_simple_gradient():
    startPoint = numpy.zeros(2, numpy.float)
    optimi = optimizer.StandardOptimizer(
        function=Quadratic(),
        step=step.GradientStep(),
        criterion=criterion.RelativeValueCriterion(0.00001),
        x0=startPoint,
        line_search=line_search.SimpleLineSearch(alpha_step=0.001))
    assert_almost_equal(optimi.optimize(),
                        numpy.array([1, 3], dtype=numpy.float))

コード例 #2

def test_simple_gradient_relative():
    startPoint = numpy.empty(2, numpy.float)
    startPoint[0] = -1.01
    startPoint[-1] = 1.01
    optimi = optimizer.StandardOptimizer(
        function=Rosenbrock(2),
        step=step.GradientStep(),
        criterion=criterion.RelativeValueCriterion(0.00001),
        x0=startPoint,
        line_search=line_search.SimpleLineSearch(alpha_step=0.001))
    assert_almost_equal(optimi.optimize(),
                        numpy.ones(2, numpy.float),
                        decimal=1)