From b39d4c569de75c50fa5221c2880ab8cbca267f02 Mon Sep 17 00:00:00 2001 From: Gustavo Rosa Date: Wed, 27 May 2020 12:49:28 -0300 Subject: [PATCH] Forgot to add the GaussianRelu examples/models script. --- examples/models/create_gaussian_relu_rbm.py | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 examples/models/create_gaussian_relu_rbm.py diff --git a/examples/models/create_gaussian_relu_rbm.py b/examples/models/create_gaussian_relu_rbm.py new file mode 100644 index 0000000..1364eae --- /dev/null +++ b/examples/models/create_gaussian_relu_rbm.py @@ -0,0 +1,5 @@ +from learnergy.models import GaussianReluRBM + +# Creates a GaussianReluRBM-based class +model = GaussianReluRBM(n_visible=784, n_hidden=128, steps=1, learning_rate=0.1, + momentum=0, decay=0, temperature=1, use_gpu=False)