1

我想检查一个用户定义的函数是否已经在 J​​uMP/julia 中注册。这是一个例子：

function foo( f, f1, f2 )

  if !function_is_registered(:f)  # This is what I'm looking for
    JuMP.register(:f,1,f1,f2)
  end
  ####
    # Optimization problem here using f
    # Leads to some return statement
  ####
end

f(x) = exp( A * x )
f1(x) = A * exp( A * x )
f2(x) = A * A * exp( A * x )
    # Function to register

A = 2
use1 = foo(f, f1, f2)
use2 = foo(f, f1, f2)
    # This second usage would fail without the check.  Can't re-register f.

从评论中应该很明显，第二次使用需要检查。据我所知，JuMP 在全局级别注册函数——一旦注册，它们就不能在本地重新定义（对吗？如果可以，这也解决了我的问题！）。

4

2 回答 2

2

这将做你想要的。

using JuMP
using Ipopt


function set_A_sol( A )
  f = (x) -> exp( A * x ) - x
  f1 = (x) -> A * exp( A * x ) - 1.0
  f2 = (x) -> A * A * exp( A * x )
  # Local redefinition of f
  try
    JuMP.register(:f, 1, f, f1, f2)
  catch e
    if e.msg == "Operator f has already been defined"
      ind = pop!( ReverseDiffSparse.univariate_operator_to_id, :f);
      deleteat!( ReverseDiffSparse.univariate_operators, ind);
      pop!( ReverseDiffSparse.user_univariate_operator_f, ind);
      pop!( ReverseDiffSparse.user_univariate_operator_fprime, ind);
      pop!( ReverseDiffSparse.user_univariate_operator_fprimeprime, ind);
      JuMP.register(:f, 1, f, f1, f2);
    end
  end
  mod = Model(solver=Ipopt.IpoptSolver(print_level=0))
  @variable(mod, - Inf <= x <= Inf )
  @NLobjective(mod, Min, f(x) )
  status=solve(mod)
  return getvalue(x)
end


julia> ans1 = set_A_sol(0.5)
1.3862943611200509


julia> ans2 = set_A_sol(1.0)
0.0


julia> ans3 = set_A_sol(2.0)
-0.34657359027997264

解释：

如果您查看在 nlp.jl 中定义的 register 函数，“注册”涉及将符号添加到字典中，保存在 ReverseDiffSparse 中。注册一个函数并手动检查这些字典以查看它们的外观。

因此，“取消注册”只涉及从所有已记录的地方删除 :f 及其派生词的所有痕迹。

于 2016-08-06T00:34:46.167 回答
0

这是基于 Tasos 建议的扩展答案（感谢 Tasos！）。

tl;dr您可以对已注册的内容使用 try-catch 语句。您还可以在全局环境中更改目标函数中的参数，但不能将它们包装在函数中。

以下有效地允许检查函数的重新定义：

function foo2( f, f1, f2 )
  try
    JuMP.register(:f,1,f1,f2)
  end
  ####
  # Optimization problem here using f
  # Leads to some return statement
  ####
  end
end

更好的是，您实际上可以使用 JuMP 寻找的简单方式f来更改目标函数中的参数（尽管您每次都需要重新定义模型，因为您无法将 a@NLparameter放入用户定义的目标中）。例如：

using JuMP
using Ipopt

f = (x) -> exp( A * x ) - x
f1 = (x) -> A * exp( A * x ) - 1.0
f2 = (x) -> A * A * exp( A * x )
    # Period objective function
JuMP.register(:f, 1, f, f1, f2)

A = 1.0
mod = Model(solver=Ipopt.IpoptSolver(print_level=0))
@variable(mod, - Inf <= x <= Inf )
@NLobjective(mod, Min, f(x) )
status=solve(mod)
println("x = ", getvalue(x))
  # Returns 0

A = 2.0
mod = Model(solver=Ipopt.IpoptSolver(print_level=0))
@variable(mod, - Inf <= x <= Inf )
@NLobjective(mod, Min, f(x) )
status=solve(mod)
println("x = ", getvalue(x))
  # Returns -0.34657 (correct)

你甚至可以重新定义f为完全不同的东西，这仍然有效。但是，您不能将其包装在函数中。例如：

function set_A_sol( A )
  f = (x) -> exp( A * x ) - x
  f1 = (x) -> A * exp( A * x ) - 1.0
  f2 = (x) -> A * A * exp( A * x )
      # Local redefinition of f
  try
    JuMP.register(:f, 1, f, f1, f2)
  end
  mod = Model(solver=Ipopt.IpoptSolver(print_level=0))
  @variable(mod, - Inf <= x <= Inf )
  @NLobjective(mod, Min, f(x) )
  status=solve(mod)
  return getvalue(x)
end

ans1 = set_A_sol(0.5)
ans2 = set_A_sol(1.0)
ans3 = set_A_sol(2.0)
  # All return 1.38629

我不完全明白为什么，但似乎第一次A设置在里面set_A_sol，JuMP 注册A一劳永逸地修复了。鉴于这是我最终想要做的事情，我仍然被困住了。欢迎提出建议！

于 2016-08-05T20:09:12.337 回答