Skip to content

How to add an optimizer ? following AbstractGPs.jl 's co2 demo. #129

@math4mad

Description

@math4mad

How to add an optimizer ? following AbstractGPs.jl 's co2 demo.

opt =Flux.Optimise.Nesterov()  #?
function agp(X,y)
    kernel =build_kernel(ParameterHandling.value(θ_init))
    for (index, num_inducing) in enumerate(Ms)
        @info "Training with $(num_inducing) points"
        m = SVGP(
            kernel, # Kernel
            GaussianLikelihood(σ), 
            AnalyticVI(),
            inducingpoints(KmeansAlg(num_inducing), X); 
            optimiser=opt, 
            Zoptimiser=false,
        )
        
        @time train!(m, X, y, 100) 
        models[index] = m 
    end
end

agp-co2-xtest-fit

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions