diff sandbox/rbm/main.py @ 400:269d5c5a4209

Cleaned up, added sparse_instance
author Joseph Turian <turian@gmail.com>
date Tue, 08 Jul 2008 23:59:57 -0400
parents 8796b91a9f09
children ffdd2c199f2a
line wrap: on
line diff
--- a/sandbox/rbm/main.py	Tue Jul 08 21:42:21 2008 -0400
+++ b/sandbox/rbm/main.py	Tue Jul 08 23:59:57 2008 -0400
@@ -1,6 +1,7 @@
 #!/usr/bin/python
 """
-    An RBM with binomial units trained with CD-1.
+Simple SGD RBM training.
+(An example of how to use the model.)
 """
 
 
@@ -22,4 +23,4 @@
     instance = nonzero_instances[i % len(nonzero_instances)]
 
     # SGD update over instance
-    model.update(instance)
+    model.update([instance])