Julia:在函数

时间:2018-06-07 10:51:53

标签: struct julia mutable

我在Julia中创建了一个简单的随机优化函数,并使用可变结构来存储参数值。基本上,我更改了可变结构中的参数,通过性能函数传递这些参数,并根据性能更新参数。

当我手动指定要更改的参数时,我有一个版本。但是,我想概括一下,以便我可以选择更改任何给定的参数。例如,将参数的可变结构表示为“args”,并且我正在更改参数“P”,我当前代码的关键部分如下所示:

args.P = P_new
loss_new = performance_function(args)
if loss_new < loss_old # performance better, keep the new value
    loss_old = loss_new
else # performance worse, go back to old value
    args.P = P_old
end

我想将其概括为一个函数:

update_parameter(args,P,performance_function)

这样我就可以使用任何一组args,在args中更改任何参数P,并且能够使用任意性能函数。为了说明args是否定义为Parameter类型,

mutable struct Parameter
    a::Float64
    b::Float64
end

args = Parameter(1,2)

然后我想使用相同的函数更新a b,

update_parameter(args,a,performance_function)
update_parameter(args,b,performance_function)

我无法弄清楚如何做到这一点。使用词典有一些聪明的方法吗?或者更好的方式满足我想要避免直接操纵可变结构的东西?感谢

1 个答案:

答案 0 :(得分:2)

您可以将setfield!与字段名一起用作Symbol

julia> mutable struct Parameter
           a::Float64
           b::Float64
       end

julia> function update!(parameters::Parameter, name::Symbol)
           setfield!(parameters, name, 5.0)
           return P
       end
update! (generic function with 1 method)

julia> P = Parameter(1, 2)
Parameter(1.0, 2.0)

julia> update!(P, :a)
Parameter(5.0, 2.0)

julia> update!(P, :b)
Parameter(5.0, 5.0)

但是,如果我正确理解您的用例,似乎字典会更简单,因为如果添加更多参数,则不必更改Parameter类型的类型定义。