[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

180 stars 26 forks source link

Closed heale04 closed 10 months ago

heale04 commented 10 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 10 months ago

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