26 lines
604 B
Python
26 lines
604 B
Python
# Common utilities for gradient descent optimization
|
|
from .functions import (
|
|
Function1D,
|
|
Function2D,
|
|
TaskFunction1D,
|
|
HimmelblauFunction,
|
|
RavineFunction,
|
|
)
|
|
from .line_search import golden_section_search, armijo_step
|
|
from .gradient_descent import gradient_descent_1d, gradient_descent_2d, heavy_ball_1d, heavy_ball_2d
|
|
|
|
__all__ = [
|
|
"Function1D",
|
|
"Function2D",
|
|
"TaskFunction1D",
|
|
"HimmelblauFunction",
|
|
"RavineFunction",
|
|
"golden_section_search",
|
|
"armijo_step",
|
|
"gradient_descent_1d",
|
|
"gradient_descent_2d",
|
|
"heavy_ball_1d",
|
|
"heavy_ball_2d",
|
|
]
|
|
|