From 2801c6c63a48195821733245c85f9a90bd21d446 Mon Sep 17 00:00:00 2001 From: Chirag Nagpal Date: Tue, 13 Apr 2021 22:36:35 +0530 Subject: [PATCH] modified: losses.py --- dsm/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsm/losses.py b/dsm/losses.py index 1cbc202..c74d744 100644 --- a/dsm/losses.py +++ b/dsm/losses.py @@ -326,7 +326,7 @@ def _weibull_pdf(model, x, t_horizon, risk='1'): lpdfs = torch.logsumexp(lpdfs, dim=1) pdfs.append(lpdfs.detach().numpy()) - return lpdfs + return pdfs def _weibull_cdf(model, x, t_horizon, risk='1'):