Changeset 172 for trunk/examples/retina/benchmark_linear.py

Timestamp:

05/12/08 18:49:09 (4 years ago)

Author:

LaurentPerrinet

Message:

more polishing to allow easier copy/paste of code

Files:

• trunk/examples/retina/benchmark_linear.py (modified) (7 diffs)

trunk/examples/retina/benchmark_linear.py

@@ -14 +14 @@
 """
 
-import os, datetime, numpy, pylab, shelve
+import os, sys, numpy, pylab, shelve
 import progressbar # see http://projects.scipy.org/pipermail/scipy-dev/2008-January/008200.html
 from NeuroTools.parameters import *
 
-N, N_exp = 100, 15
+N, N_exp = 1000, 6
 p = ParameterSet({})
-t_smooth = 10. # ms.  integration time to show fiber activity
+t_smooth = 100. # ms.  integration time to show fiber activity
+from retina import *
+retina = Retina(N)
 
-results = shelve.open('results/benchmark_linear')
+name = sys.argv[0].split('.')[0] # name of the current script withpout the '.py' part
+results = shelve.open('results/mat-' + name)
 try:
     #boing # to force recomputing
@@ -30 +33 @@
 
 except:
-    from retina import *
-
-    retina = Retina(N)
     snr  = retina.params['snr']* numpy.linspace(0.1,2.0,N_exp)
     p.snr = ParameterRange(list(snr))
@@ -44 +44 @@
     temporal_ON, temporal_OFF = [],[]
 
-    pbar=progressbar.ProgressBar(widgets=["calculating", " ", progressbar.Percentage(), ' ',
+    pbar=progressbar.ProgressBar(widgets=[name, " ", progressbar.Percentage(), ' ',
             progressbar.Bar(), ' ', progressbar.ETA()], maxval=N_exp)
     for i_exp,experiment in enumerate(p.iter_inner()):
@@ -58 +58 @@
         pbar.update(i_exp)
 
-
-    results = retina.params
+    
     results['lower_edges'] = lower_edges
     results['temporal_ON'] = temporal_ON
@@ -66 +65 @@
     pbar.finish()
 
+results.close()
 
 ###############################################################################
@@ -88 +88 @@
 #pylab.subplot(121)
 #pylab.title('time course')# ON
-pylab.xticks( numpy.round(numpy.linspace(0, results.simtime, 5),0) )
+pylab.xticks( numpy.round(numpy.linspace(0, retina.params.simtime, 5),0) )
 pylab.ylabel('Firing frequency (Hz)')
 pylab.xlabel('time (ms)')
@@ -103 +103 @@
 #pylab.legend()
 
-pylab.savefig('results/benchmark_linear.pdf') #, dpi=300) #
-#pylab.savefig('results/benchmark_linear.png', dpi=300) #
+pylab.savefig('results/fig-' + name + '.pdf')
+pylab.savefig('results/fig-' + name + '.png', dpi = 300)
 
-results.close()
+#TODO: make a plot showing it's the right point process with a histogram