Forgot to add the GaussianRelu examples/models script.

gugarosa · May 27, 2020 · b39d4c5 · b39d4c5
1 parent e60ff14
commit b39d4c5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/examples/models/create_gaussian_relu_rbm.py 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)