Commits

Olivier Grisel committed 83074b2

fix xp_movie_prediction.py dataset construction

  • Participants
  • Parent commits 94b89af
  • Branches pysgd-arrays

Comments (0)

Files changed (1)

File experiments/xp_movie_prediction.py

 
     # load the content of the movie data into a single layered brain so as to
     # use it's API to build training data suitable for evaluating algorithms
-    for i in xrange(CYCLES):
+    for i in xrange(BATCH_SIZE):
         ma.next()
         #print "## training cycle #%d" % i
         #ma.train()
         #ma.plot_mse_gain_over_identity("gain_mse.png")
 
     end_generation = time.time()
-    log.info("generated %d frames in %03fs",
+    log.info("loaded %d frames in the brain in %03fs",
              BATCH_SIZE, end_generation - start)
 
     # dump the data set on disk so as to be able to quickly reuse it later