Changeset 166 for trunk/examples/retina/benchmark_noise.py

Timestamp:

05/07/08 18:56:23 (4 years ago)

Author:

LaurentPerrinet

Message:

Code Jam #2 : retina/benchmark_noise and benchmark_linear should now work

Files:

• trunk/examples/retina/benchmark_noise.py (modified) (5 diffs)

trunk/examples/retina/benchmark_noise.py

@@ -13 +13 @@
 """
 
-import os, datetime, numpy
-import retina as model
+import os, datetime, numpy, shelve
 
 from NeuroTools.parameters import *
@@ -22 +21 @@
 import progressbar # see http://projects.scipy.org/pipermail/scipy-dev/2008-January/008200.html
 
+N, N_exp_noise = 100, 22
 p = ParameterSet({})
-N, N_exp_noise = 10, 22
 p.noise_std = ParameterRange(list(10.**(numpy.linspace(-.50,1.,N_exp_noise))))
-retina = model.Retina(N)
-retina.params['snr'] = 0
 
 results = shelve.open('results/benchmark_noise')
@@ -32 +29 @@
     CRF = results['CRF']
 except:
+    import retina as model
 
+
+    retina = model.Retina(N)
+    retina.params['snr'] = 0
     # calculates the dimension of the parameter space
     results_dim, results_label = p.parameter_space_dimension_labels()
@@ -50 +51 @@
         index = p.parameter_space_index(experiment)
         # put the data at the right position in the results array
-        CRF[index] = data.mean_rate()#
+        CRF[index] = data['out_ON_DATA'].mean_rate()#
         pbar.update(i_exp)
 
@@ -80 +81 @@
 #pylab.axes([Lmargin, dmargin , 1.0 - Rmargin- Lmargin,1.0-umargin-dmargin]) # [left, bottom, width, height]
 
-idx_ = numpy.argsort(noise_std) # extract lines
-pylab.plot(noise_std[idx_],CRF[idx_],'go-', label='line 1', linewidth=2)
+pylab.plot(p.noise_std._values,CRF,'go-', label='line 1', linewidth=2)
 pylab.ylabel('Firing Frequency (Hz)')
 pylab.xlabel('Noise amplitude')