add show method

Author: CarloLucibello

src/interface.jl

@@ -281,7 +281,17 @@ macro def(expr)
   # resulting structs to be @doc... cannot if macro returns a block.)
   kwmethod = :($rule(; $(params...)) = $rule($(names...)))
   push!(lines, positional, kwmethod)
-  return esc(expr)
+  # return esc(expr)
+# end
+
+  return quote 
+          Base.@__doc__ $expr
+
+          function Base.show(io::IO, r::$rule)
+            pairs = ["$n=$(getfield(r, n))" for n in [($names...)]]
+            print(io, $rule,"(", join(pairs, ", "), ")")
+          end
+        end |> esc
 end
 
 _def_typeof(val) = typeof(val)

src/rules.jl

@@ -148,9 +148,7 @@ function adjust(r::RMSProp; kw...)
 end
 
 function Base.show(io::IO, o::RMSProp)
-  print(io, "RMSProp(")
-  join(io, [o.eta, o.rho, o.epsilon], ", ")
-  print(io, "; centred = ", o.centred, ")")
+  print(io, "RMSProp(eta=$(o.eta), rho=$(o.rho), epsilon=$(o.epsilon), centred=$(o.centred)")
 end
 
 
@@ -565,6 +563,11 @@ function apply!(o::AdamW, state, x::AbstractArray{T}, dx) where T
   return (mt, vt, βt .* β), dx′′
 end
 
+function Base.show(io::IO, o::AdamW)
+  print(io, "AdamW(eta=$(o.eta), beta=$(o.beta), lambda=$(o.lambda), epsilon=$(o.epsilon), couple=$(o.couple))")
+end
+
+
 """
     AdaBelief(η = 0.001, β = (0.9, 0.999), ϵ = 1e-16)
     AdaBelief(; [eta, beta, epsilon])