def __csd_correlation(v, m): '''compute correlation coefficient between CSD estimate and CSD for a given source diameter''' if m == 'delta': icsd_input[m].update({'diam': v * pq.m}) _icsd = icsd.DeltaiCSD(**icsd_input[m]) corrcoef = pl.corrcoef( CSD_filtered.flatten(), pl.array(_icsd.filter_csd(_icsd.get_csd()) / pq.m).flatten()) elif m == 'step': icsd_input[m].update({'diam': v * pq.m}) _icsd = icsd.StepiCSD(**icsd_input[m]) corrcoef = pl.corrcoef( CSD_filtered.flatten(), pl.array(_icsd.filter_csd(_icsd.get_csd()) / pq.m).flatten()) elif m == 'spline': icsd_input[m].update({'diam': v * pq.m}) _icsd = icsd.SplineiCSD(**icsd_input[m]) corrcoef = pl.corrcoef( CSD76ptF.flatten(), pl.array(_icsd.filter_csd(_icsd.get_csd()) / pq.m).flatten()) else: raise Exception, 'm = %s should be either [delta, step, spline]' % m return corrcoef[0, -1]
def test_StepiCSD_units_04(self): '''test non-continous z_j array''' # set some parameters for ground truth csd and csd estimates., e.g., # we will use same source diameter as in ground truth # contact point coordinates z_j = np.arange(21) * 1E-4 * pq.m # source coordinates z_i = z_j # current source density magnitude C_i = np.zeros(z_i.size) * pq.A / pq.m**3 C_i[7:12:2] += np.array([-.5, 1., -.5]) * pq.A / pq.m**3 # source radius (delta, step) R_i = np.ones(z_i.size) * 1E-3 * pq.m # source height (cylinder) h_i = np.ones(z_i.size) * 1E-4 * pq.m # conductivity, use same conductivity for top layer (z_j < 0) sigma = 0.3 * pq.S / pq.m sigma_top = sigma # flag for debug plots plot = False # get LFP and CSD at contacts phi_j, C_i = get_lfp_of_cylinders(z_j, z_i, C_i, R_i, h_i, sigma, plot) inds = np.delete(np.arange(21), 5) step_input = { 'lfp': phi_j[inds], 'coord_electrode': z_j[inds], 'diam': R_i[inds] * 2, 'sigma': sigma, 'sigma_top': sigma, 'h': h_i[inds], 'tol': 1E-12, # Tolerance in numerical integration 'f_type': 'gaussian', 'f_order': (3, 1), } step_icsd = icsd.StepiCSD(**step_input) csd = step_icsd.get_csd() self.assertEqual(C_i.units, csd.units) nt.assert_array_almost_equal(C_i[inds], csd)
def __csd_error(v, m): '''using squared difference summed''' if m == 'delta': icsd_input[m].update({'diam': v * pq.m}) _icsd = icsd.DeltaiCSD(**icsd_input[m]) error = ( (pl.array(_icsd.filter_csd(_icsd.get_csd()) / (100E-6 * pq.m)).reshape(CSD_filtered.size) * 1E-9 - CSD_filtered.reshape(CSD_filtered.size))**2).sum() elif m == 'step': icsd_input[m].update({'diam': v * pq.m}) _icsd = icsd.StepiCSD(**icsd_input[m]) error = ((pl.array(_icsd.filter_csd(_icsd.get_csd())).reshape( CSD_filtered.size) * 1E-9 - CSD_filtered.reshape(CSD_filtered.size))**2).sum() elif m == 'spline': icsd_input[m].update({'diam': v * pq.m}) _icsd = icsd.SplineiCSD(**icsd_input[m]) error = ((pl.array(_icsd.filter_csd(_icsd.get_csd())).reshape( CSD76ptF.size) * 1E-9 - CSD76ptF.reshape(CSD76ptF.size))**2).sum() else: raise Exception, 'm = %s should be either [delta, step, spline]' % m return error
_icsd.filter_csd(_icsd.get_csd()) / (100E-6 * pq.m) * 1E-9 }) my_errors['delta'], my_diams['delta'], diam_best['delta'] = \ minimize_icsd_error_brute(m) icsd_input[m].update({'diam': diam_best['delta'] * pq.m}) print 'best diameter delta: %.5e' % diam_best['delta'] _icsd = icsd.DeltaiCSD(**icsd_input[m]) icsd_output.update({ 'icsd_delta': _icsd.filter_csd(_icsd.get_csd()) / (100E-6 * pq.m) * 1E-9 }) elif m == 'step': _icsd = icsd.StepiCSD(**icsd_input[m]) icsd_output.update( {'icsd_step': _icsd.filter_csd(_icsd.get_csd()) * 1E-9}) my_errors['step'], my_diams['step'], diam_best['step'] = \ minimize_icsd_error_brute(m) icsd_input[m].update({'diam': diam_best['step'] * pq.m}) print 'best diameter step: %.5e' % diam_best['step'] _icsd = icsd.StepiCSD(**icsd_input[m]) icsd_output.update( {'icsd_step': _icsd.filter_csd(_icsd.get_csd()) * 1E-9}) elif m == 'spline': _icsd = icsd.SplineiCSD(**icsd_input[m])