TRAIS-Lab / dattri

`dattri` is a PyTorch library for developing, benchmarking, and deploying efficient data attribution algorithms.
https://trais-lab.github.io/dattri/
MIT License
28 stars 8 forks source link

LDS corrections #31

Closed jackhaohuang closed 4 months ago

jackhaohuang commented 4 months ago

Description

1. Motivation and Context

We complete the proposal in the issue #9.

2. Summary of the change

We implemented functions retrain_lds, calculate_lds_groundtruth, and lds. We also designed unit-tests for all three functions.

3. What tests have been added/updated for the change?