Add visualization tooling / plot recipes
Nobody has claimed this yet.
- Dominant language
- Julia
- Stars
- 839
- Forks
- 100
- Avg merge
- 20h 43m
- Merged PRs (30d)
- 38
Description
FluxOptTools allows visualizing loss when using Optim.jl. There's also some integration between Optim and TensorBoardLogger. Neither of these integrations can be directly used when using GalacticOptim in place of Optim. It seems like a first step would be add similar integrations / recipes for GalacticOptim.
As a first pass, I got TensorBoardLogger to plot loss with the following callback:
function make_tb_cb(name="")
logger = TBLogger(logdir, tb_overwrite)
function callback(p, l)
with_logger(logger) do
@info "$(name)" loss=l
end
return false
end
return callback
end
maxiters=100
GalacticOptim.solve(prob, Optim.BFGS(); cb=make_tb_cb("BFGS-$(maxiters)"), maxiters=maxiters)
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reviewing the GalacticOptim.solve callback shown in the issue, then compare the visualization integrations in FluxOptTools and TensorBoardLogger. Define which plotting recipes or integrations GalacticOptim should support and verify that loss can be visualized through the resulting interface.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- julia
- Domain
- data-visualization
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100