[added feature] regularization term for inv(xtx) calculation

MadryLab / trak

A fast, effective data attribution method for neural networks in PyTorch

https://trak.csail.mit.edu/

MIT License

169 stars 22 forks source link

Closed heale04 closed 7 months ago

heale04 commented 7 months ago

added a regularization term argument for traker that alters the calculation of inv(xtx) for when it is not invertible.

kristian-georgiev commented 7 months ago

Thank you @heale04, it's very useful to have ridge regularization for the inverse!