Revert "Added Dense and Conv BatchEnsemble layers along with unit tes…

…ts and example on MNIST classification using LeNet5 (#4)" This reverts commit ed3d16c.
aced-differentiate · Sep 7, 2021 · 683827b · 683827b
1 parent ed3d16c
commit 683827b
Show file tree

Hide file tree

Showing 11 changed files with 18 additions and 728 deletions.
diff --git a/Project.toml b/Project.toml
@@ -10,7 +10,6 @@ Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 ReliabilityDiagrams = "e5f51471-6270-49e4-a15a-f1cfbff4f856"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
-Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]
 julia = "1"

diff --git a/examples/batchensemble.jl b/examples/batchensemble.jl
diff --git a/src/DeepUncertainty.jl b/src/DeepUncertainty.jl
@@ -1,17 +1,10 @@
 module DeepUncertainty
 
-using Flux
-using Random
-using Flux: @functor, glorot_normal, create_bias
-
 # Export layers 
 export MCLayer, MCDense, MCConv
-export DenseBatchEnsemble, ConvBatchEnsemble
 export mean_loglikelihood, brier_score, ExpectedCalibrationError, prediction_metrics
 
 include("metrics.jl")
 include("layers/mclayers.jl")
-include("layers/BatchEnsemble/dense.jl")
-include("layers/BatchEnsemble/conv.jl")
 
 end
diff --git a/src/layers/BatchEnsemble/conv.jl b/src/layers/BatchEnsemble/conv.jl