def sample_from_network_hawkes(K, T, dt, dt_max, B): # Create a true model true_model = DiscreteTimeNetworkHawkesModelSpikeAndSlab(K=K, dt=dt, dt_max=dt_max, B=B, network_hypers=dict(p=0.1)) # Plot the true network plt.ion() true_model.plot_network() # Sample from the true model S,R = true_model.generate(T=T) # Return the spike count matrix return S, true_model