Skip to content

Commit 7a7d165

Browse files
authored
Update README.md
1 parent 5bf9b15 commit 7a7d165

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

README.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,9 +112,10 @@ Useful Tips
112112
Checklist before Running Experiments:
113113
- [ ] Your data should have binary labels **0,1** and **1** is the **minority class** and **0** is the **majority class**
114114
- [ ] Compute the **imbalance_ratio** from your train set and pass it to `AUCMLoss(imratio=xxx)`
115-
- [ ] Choose a proper **initial learning rate**, e.g., **lr=[0.1, 0.05]** usually works better
115+
- [ ] Adopt a proper **initial learning rate**, e.g., **lr=[0.1, 0.05]** usually works better
116+
- [ ] Choose `libauc.optimizers.PESG` to optimize `AUCMLoss(imratio=xxx)`
116117
- [ ] Use `optimizer.update_regularizer(decay_factor=10)` to update learning rate and regularizer in stagewise
117-
- [ ] Use activation function, e.g., `torch.sigmoid()`, before passing model outputs to loss function
118+
- [ ] Add activation layer, e.g., `torch.sigmoid(logits)`, before passing model outputs to loss function
118119
- [ ] Reshape both variables **preds** and **targets** to `(N, 1)` before calling loss function
119120

120121

0 commit comments

Comments
 (0)