Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion open_eeg_bench/ridge_probe.py
Original file line number Diff line number Diff line change
Expand Up @@ -246,7 +246,8 @@ def _fit_streaming_ridge(
) # tensor of shape (K,)

# Among tied best scores, pick the largest λ (most regularization):
# more parsimonious, fewer effective degrees of freedom, numerically stabler.
# when val cannot discriminate (e.g. discretized scores on small val sets,
# plateaus), the most regularized solution is the safest bet.
# NaN-aware: a NaN val score means the val pass hit non-finite features
# or predictions — exclude it from both max and tie-breaking.
finite_mask = torch.isfinite(val_scores)
Expand Down
Loading