plot_head_direction_input.py 822 B

123456789101112131415161718192021
  1. import matplotlib.pyplot as plt
  2. import numpy
  3. from brian2.units import *
  4. from scripts.spatial_network.head_direction_index_over_noise_scale import create_head_direction_input
  5. tunings = numpy.linspace(-numpy.pi, numpy.pi, 100, endpoint=False)
  6. baseline = 0.05 * nA
  7. amplitude = 0.6 * nA
  8. sharpness = 2.5
  9. direction = 0
  10. input_to_excitatory_population = create_head_direction_input(baseline, tunings,
  11. sharpness,
  12. amplitude, direction)
  13. plt.plot(tunings / numpy.pi * 180, input_to_excitatory_population / nA)
  14. plt.xlabel("$\\theta-\\theta_p$")
  15. plt.ylabel("I (nA)")
  16. plt.ylim(0, numpy.max(input_to_excitatory_population / nA) * 1.05)
  17. plt.savefig('../../figures/evolve_spatial_layout/broad_input.png', dpi=300)