wassname/torch-neuralpointprocess
1
0
Fork 0
mirror of https://github.com/wassname/torch-neuralpointprocess.git synced 2026-09-09 11:38:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

use nb

Browse Source
This commit is contained in:
wassname
2022-02-11 15:36:01 +08:00
parent c0cfe2186e
commit e8ab8fc1f4
8 changed files with 1422 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils.py
+5 -6
View File
@@ -56,10 +56,8 @@ def plt_lmbda(timeseries, model, seq_len, log_mode=False, dt=0.01, lmbda0=0.2, a
# exponential_hwakes : lmbda0, alpha, beta: 0.2, 0.8, 1.0
# lmbda = lambda0 + alpha*sum(exp{-beta*(t-t_i)})
lmbda_dict[0] = np.zeros(t_span.shape)
for t, e in timeseries:
target = (t_span > t)
lmbda_dict[0][target] += alpha*np.exp(-beta*(t_span[target]-t))
@@ -71,7 +69,8 @@ def plt_lmbda(timeseries, model, seq_len, log_mode=False, dt=0.01, lmbda0=0.2, a
_, _, _, pred_dict[0] = model((test_timeseq, test_eventseq))
plt.plot(t_span, lmbda_dict[0], color='green')
plt.plot([t for t, e in timeseries][seq_len-1:], np.array(pred_dict[0].detach()), color='olive')
plt.scatter([t for t, e in timeseries], [-1 for _ in timeseries], color='blue')
plt.show()
plt.plot(t_span, lmbda_dict[0], color='green', label='true prob')
plt.plot([t for t, e in timeseries][seq_len-1:], np.array(pred_dict[0].detach()), color='olive', label='pred prob')
plt.scatter([t for t, e in timeseries], [-1 for _ in timeseries], color='blue', label='events')
plt.legend()
plt.show()