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

Timestamp:

11/12/08 19:49:46 (4 years ago)

Author:

LaurentPerrinet

Message:

wiki:examples : reflect changes in the wiki:parameters API / polishing figures to give usable examples

Files:

• trunk/examples/retina/benchmark_linear.py (modified) (3 diffs, 1 prop)

trunk/examples/retina/benchmark_linear.py

@@ -19 +19 @@
 
 N, N_exp = 1000, 6
-p = ParameterSet({})
 t_smooth = 100. # width (in ms) of the integration window
 from retina import *
 retina = Retina(N)
+retina.params['amplitude'] = 1
+snr  = retina.params['snr']* numpy.linspace(0.1,2.0,N_exp)
+p =  ParameterSpace({'snr' : ParameterRange(list(snr))})
+
 
 name = sys.argv[0].split('.')[0] # name of the current script withpout the '.py' part
@@ -33 +36 @@
 
 except:
-    snr  = retina.params['snr']* numpy.linspace(0.1,2.0,N_exp)
-    p.snr = ParameterRange(list(snr))
+
     # calculates the dimension of the parameter space
     results_dim, results_label = p.parameter_space_dimension_labels()
@@ -78 +80 @@
 
 """
-pylab.close('all')
-pylab.rcParams.update(pylab_params(fig_width_pt = 497.9/2., ratio = 1.))
-pylab.figure(num = 1, dpi=150, facecolor='w', edgecolor='k')
-Lmargin, Rmargin, dmargin, umargin = 0.2, 0.05, 0.15,  0.05
-pylab.axes([Lmargin, dmargin , 1.0 - Rmargin- Lmargin,1.0-umargin-dmargin]) # [left, bottom, width, height]
+#pylab.close('all')
+#pylab.rcParams.update(pylab_params(fig_width_pt = 497.9/2., ratio = 1.))
+pylab.figure(1)
+fmax = numpy.max([numpy.max(temporal_OFF[:]),numpy.max(temporal_ON[:])])
 
+pylab.subplot(211)
 for i_exp in range(N_exp):
-    pylab.plot(lower_edges,  # TODO add a half bin
-                    temporal_ON[i_exp])#,
-                       # label= 'snr=' + str(benchmark.get('experiments')[experiment]['snr']))
-#pylab.subplot(121)
-#pylab.title('time course')# ON
+    pylab.plot(lower_edges[:-1] + t_smooth/2,  # TODO add a half bin
+                    temporal_ON[i_exp])#
 pylab.xticks( numpy.round(numpy.linspace(0, retina.params.simtime, 5),0) )
-pylab.ylabel('Firing frequency (Hz)')
+pylab.ylabel('ON Firing frequency (Hz/neuron)')
+pylab.axis([0, retina.params.simtime, 0.0, fmax])
+pylab.subplot(212)
+for i_exp in range(N_exp):
+    pylab.plot(lower_edges[:-1] + t_smooth/2,  # TODO add a half bin
+                    temporal_OFF[i_exp],
+                        label= 'snr= %5.3f' % p.snr._values[i_exp] )
+pylab.xticks( numpy.round(numpy.linspace(0, retina.params.simtime, 5),0) )
+pylab.ylabel('OFF Firing frequency (Hz/neuron)')
 pylab.xlabel('time (ms)')
-pylab.axis('tight')
+pylab.axis([0, retina.params.simtime, 0.0, fmax ])
 
-#pylab.legend()
+pylab.legend(loc='upper right')
 
-#pylab.subplot(122)
-#pylab.title('time course OFF')
-#pylab.xticks( numpy.linspace(0, simtime, 5) )
-#pylab.ylabel('firing frequency (Hz)')
-#pylab.axis('tight')
-#pylab.xlabel('time (ms)')
-#pylab.legend()
 
-pylab.savefig('results/fig-' + name + '.pdf')
-pylab.savefig('results/fig-' + name + '.png', dpi = 300)
+if 0:
+    pylab.ion()
+    #pylab.show()
+else:
+    pylab.savefig('results/fig-' + name + '.pdf')
+    pylab.savefig('results/fig-' + name + '.png', dpi = 300)
 
-#TODO: make a plot showing it's the right point process with a histogram