Skip to content

Commit ab89d82

Browse files
authored
Revert "Added Dense and Conv BatchEnsemble layers along with unit tests and example on MNIST classification using LeNet5 (#4)" (#14)
This reverts commit ed3d16c.
1 parent ed3d16c commit ab89d82

File tree

11 files changed

+18
-728
lines changed

11 files changed

+18
-728
lines changed

Project.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
1010
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
1111
ReliabilityDiagrams = "e5f51471-6270-49e4-a15a-f1cfbff4f856"
1212
Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
13-
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
1413

1514
[compat]
1615
julia = "1"

examples/batchensemble.jl

Lines changed: 0 additions & 207 deletions
This file was deleted.

src/DeepUncertainty.jl

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,10 @@
11
module DeepUncertainty
22

3-
using Flux
4-
using Random
5-
using Flux: @functor, glorot_normal, create_bias
6-
73
# Export layers
84
export MCLayer, MCDense, MCConv
9-
export DenseBatchEnsemble, ConvBatchEnsemble
105
export mean_loglikelihood, brier_score, ExpectedCalibrationError, prediction_metrics
116

127
include("metrics.jl")
138
include("layers/mclayers.jl")
14-
include("layers/BatchEnsemble/dense.jl")
15-
include("layers/BatchEnsemble/conv.jl")
169

1710
end

src/layers/BatchEnsemble/conv.jl

Lines changed: 0 additions & 145 deletions
This file was deleted.

0 commit comments

Comments
 (0)