【发布时间】:2015-03-24 21:11:58
【问题描述】:
我对循环函数的循环结果有疑问。它只在循环内计数一次,并为第一个 raw 选择最佳解决方案,然后停止。
我想记住矩阵zmienne 每一行的最佳解决方案。我做错了什么?
schaffer <- function(xx)
{x1 <- xx[1]
x2 <- xx[2]
fact1 <- (sin(x1^2-x2^2))^2 - 0.5
fact2 <- (1 + 0.001*(x1^2+x2^2))^2
y <- 0.5 + fact1/fact2
return(y)
}
gradient_descent <- function(func, step, niter) {
N <- 3 #N- number of random points
zmienne <- matrix(runif(N*2, min = -100, max = 100), N, 2)
print(zmienne)
h = 0.001;
iter_count = 0;
for (i in 1:N) {
x_0 <- zmienne[i,]
x_n = x_0;
for (j in 1:niter) {
func_grad = (func(x_n+h) - func(x_n))/h;
if (abs(func_grad) < 0.0001) { break; }
x_n = x_n - step * func_grad;
print(x_n)
iter_count = iter_count + 1
}
}
return(list(iterations = niter, best_value = func_grad, best_state = x_n, x0=x_0))
}
solution_m1 <- gradient_descent(schaffer, 0.1, 20)
solution_m1
【问题讨论】:
-
您能否提供您的数据(或子集)以便我们测试输出?
-
考虑在第一个循环之外分配
iterations、best_value、best_state和x0和NA,并将最佳结果添加到每行的这些变量中。然后在外部循环之外返回一个由iterations、best_value、best_state和x0组成的列表。 -
你能通过在代码中添加你的建议来解释我的修改吗?我应该在关闭内部循环后添加return,然后在关闭主循环后添加?
标签: r loops for-loop gradient-descent