Update autoPyTorch/utils/implementations.py

Co-authored-by: nabenabe0928 <47781922+nabenabe0928@users.noreply.github.com>
automl · Jun 23, 2021 · 0d1e50a · 0d1e50a
1 parent f9a5704
commit 0d1e50a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/autoPyTorch/utils/implementations.py b/autoPyTorch/utils/implementations.py
@@ -52,6 +52,7 @@ def __call__(self, y: Union[np.ndarray, torch.Tensor]) -> np.ndarray:
             y = y.detach().cpu().numpy() if y.is_cuda else y.numpy()
         counts_one = np.sum(y, axis=0)
         counts_zero = y.shape[0] - counts_one
+        # pos_weight for torch.nn.BCEWithLogitsLoss
         weights = counts_zero / np.maximum(counts_one, 1)
 
         return np.array(weights)