From 84d77f45857909e496e219e1af29cef07da692d2 Mon Sep 17 00:00:00 2001
From: Wendi Li <wendili.academic@qq.com>
Date: Sat, 23 Jan 2021 13:21:46 +0000
Subject: [PATCH] Update pytorch_nn.py

---
 qlib/contrib/model/pytorch_nn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/qlib/contrib/model/pytorch_nn.py b/qlib/contrib/model/pytorch_nn.py
index a41eeabbb..1623e7e1c 100644
--- a/qlib/contrib/model/pytorch_nn.py
+++ b/qlib/contrib/model/pytorch_nn.py
@@ -259,7 +259,7 @@ class DNNModelPytorch(Model):
             loss = torch.mul(sqr_loss, w).mean()
             return loss
         elif loss_type == "binary":
-            loss = nn.BCELoss()
+            loss = nn.BCELoss(weight=w)
             return loss(pred, target)
         else:
             raise NotImplementedError("loss {} is not supported!".format(loss_type))