Skip to content

Commit 5587bb4

Browse files
committed
add citation
1 parent af879fa commit 5587bb4

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

encoding/models/ridge_regression.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
import logging
44
from encoding.models.ridge_utils import svd_wrapper, mult_diag, z_score
55

6+
# this is a torch implementation of the following: https://github.com/HuthLab/encoding-model-scaling-laws/blob/main/ridge_utils/ridge.py
7+
68

79
def ridge_torch(
810
Rstim: torch.Tensor,

0 commit comments

Comments
 (0)