mirror of
https://github.com/gonum/gonum.git
synced 2025-10-02 22:12:14 +08:00
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
// Copyright ©2016 The Gonum Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package optimize_test
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"gonum.org/v1/gonum/optimize"
|
|
"gonum.org/v1/gonum/optimize/functions"
|
|
)
|
|
|
|
func ExampleLocal() {
|
|
p := optimize.Problem{
|
|
Func: functions.ExtendedRosenbrock{}.Func,
|
|
Grad: functions.ExtendedRosenbrock{}.Grad,
|
|
}
|
|
|
|
x := []float64{1.3, 0.7, 0.8, 1.9, 1.2}
|
|
settings := optimize.DefaultSettings()
|
|
settings.Recorder = nil
|
|
settings.GradientThreshold = 1e-12
|
|
settings.FunctionConverge = nil
|
|
|
|
result, err := optimize.Local(p, x, settings, &optimize.BFGS{})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err = result.Status.Err(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
fmt.Printf("result.Status: %v\n", result.Status)
|
|
fmt.Printf("result.X: %v\n", result.X)
|
|
fmt.Printf("result.F: %v\n", result.F)
|
|
fmt.Printf("result.Stats.FuncEvaluations: %d\n", result.Stats.FuncEvaluations)
|
|
// Output:
|
|
// result.Status: GradientThreshold
|
|
// result.X: [1 1 1 1 1]
|
|
// result.F: 0
|
|
// result.Stats.FuncEvaluations: 35
|
|
}
|