JuliaTrustworthyAI / LaplaceRedux.jl

Effortless Bayesian Deep Learning through Laplace Approximation for Flux.jl neural networks.
https://www.taija.org/LaplaceRedux.jl/
MIT License
39 stars 3 forks source link

Laplace input format #91

Closed pasq-cat closed 4 months ago

pasq-cat commented 4 months ago

I have realized that Laplace has no problem to deal with input in the format

Base.Iterators.Zip{Tuple{Vector{Vector{Float32}}, Vector{Float32}}}

but it doesn't work if the data are in the format

Base.Iterators.Zip{Tuple{Matrix{Float64}, Vector{Float64}}}

to avoid errors i had to use zip(eachrow(X),y) and not use the dataloader, which may be convenient when the dataset is big.