Added pip command to install SBI

Signed-off-by: samadpls <[email protected]>

examples/howto/sbi_hnn_core_parameter_inference.py

@@ -15,6 +15,8 @@
 #          Mainak Jas <[email protected]>
 
 ###############################################################################
+# First, we need to import the necessary packages. The Simulation-Based
+# Inference (SBI) package can be installed with ``pip install sbi``.
 # Let us import ``hnn_core`` and all the necessary libraries.
 
 import numpy as np
@@ -62,7 +64,7 @@ def set_params(param_values, net=None):
     'weight_pyr': val.tolist()
 }
 
-net = jones_2009_model(mesh_shape=(1, 1))
+net = jones_2009_model(mesh_shape=(3, 3))
 batch_simulator = BatchSimulate(set_params=set_params,
                                 net=net,
                                 tstop=170)
@@ -140,7 +142,6 @@ def simulator_batch(param):
 plt.axvline(unknown_param.item(), color='r', linestyle='dashed',
             linewidth=2, label='True Parameter')
 plt.legend()
-plt.savefig('posterior_distribution_log.png')
 plt.show()
 
 ###############################################################################
@@ -158,7 +159,6 @@ def simulator_batch(param):
 plt.xlabel('True Parameter')
 plt.ylabel('Inferred Parameter')
 plt.title('SBI Performance on Unseen Data')
-plt.savefig('sbi_performance_unseen.png')
 plt.show()
 
 ###############################################################################