0.618算法和基于Armijo准则的线搜索回退法

0.618代码如下:

import math

# 定义函数 h(t) = t^3 - 2t + 1

def h(t):

return t**3 - 2*t + 1

# 0.618 算法

def golden_section_search(a, b, epsilon):

ratio = 0.618

while (b - a) > epsilon:

x1 = b - ratio * (b - a)

x2 = a + ratio * (b - a)

h_x1 = h(x1)

h_x2 = h(x2)

if h_x1 < h_x2:

b = x2

else:

a = x1

return a # 或者返回 b ,因为它们的值非常接近

# t 大于等于 0 的范围内进行搜索

t_min_618 = golden_section_search(0, 3, 0.001)

print("0.618 算法找到的最小值: ", h(t_min_618))

基于Armijo准则的线搜索回退法代码如下:

import numpy as np

def h(t):

return t**3 - 2*t + 1

def h_derivative(t):

return 3*t**2 - 2

def armijo_line_search(t_current, direction, alpha, beta, c1):

t = t_current

step_size = 1.0

while True:

if h(t + direction * step_size) <= h(t) + alpha * step_size * direction * h_derivative(t):

return t + direction * step_size

else:

step_size *= beta

if np.abs(step_size) < 1e-6:

break

return None

def gradient_descent(start, end, alpha, beta, c1, epsilon):

t = start

while True:

if t > end:

break

direction = -h_derivative(t) # 负梯度方向

next_t = armijo_line_search(t, direction, alpha, beta, c1)

if next_t is None or np.abs(h_derivative(next_t)) <= epsilon:

return next_t

t = next_t

return None

# 参数设置

alpha = 0.1 # Armijo 准则中的参数 alpha

beta = 0.5 # Armijo 准则中的参数 beta

c1 = 1e-4 # 自定义参数,用于控制 Armijo 条件的满足程度

epsilon = 1e-6 # 梯度范数的终止条件

# 搜索区间为 [0,3]

start = 0

end = 3

# 执行梯度下降算法,求得近似最小值点

t_min = gradient_descent(start, end, alpha, beta, c1, epsilon)

print(" 求得的最小值点为 :", t_min)

print(" 最小值点的函数值为 :", h(t_min))

相关推荐
星星的月亮叫太阳11 分钟前
large-scale-DRL-exploration 代码阅读 总结
python·算法
喜欢吃豆25 分钟前
GraphRAG 技术教程:从核心概念到高级架构
人工智能·架构·大模型
王哈哈^_^25 分钟前
YOLOv11视觉检测实战:安全距离测算全解析
人工智能·数码相机·算法·yolo·计算机视觉·目标跟踪·视觉检测
..Cherry..38 分钟前
Etcd详解(raft算法保证强一致性)
数据库·算法·etcd
AI浩44 分钟前
FeatEnHancer:在低光视觉下增强目标检测及其他任务的分层特征
人工智能·目标检测·目标跟踪
深度学习lover1 小时前
<数据集>yolo航拍交通目标识别数据集<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·航拍交通目标识别
商汤万象开发者1 小时前
LazyLLM教程 | 第13讲:RAG+多模态:图片、表格通吃的问答系统
人工智能·科技·算法·开源·多模态
IT管理圈1 小时前
AI agent正在重塑组织:麦肯锡的“智能体组织“解读
人工智能
YuanDaima20481 小时前
[CrewAI] 第5课|基于多智能体构建一个 AI 客服支持系统
人工智能·笔记·多智能体·智能体·crewai