deploy: 301cde7

TRAIS-Lab · Sep 6, 2024 · b2af5be · b2af5be
1 parent e95dcb8
commit b2af5be
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/.doctrees/api/benchmark.doctree b/.doctrees/api/benchmark.doctree
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
diff --git a/api/benchmark.html b/api/benchmark.html
@@ -562,7 +562,7 @@ <h1>Benchmark Functions<a class="headerlink" href="#benchmark-functions" title="
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
 <dd class="field-odd"><ul class="simple">
 <li><p><strong>score</strong> (<em>torch.Tensor</em>) – The data attribution score tensor with the shape
-(num_test_samples, num_train_samples).</p></li>
+(num_train_samples, num_test_samples).</p></li>
 <li><p><strong>ground_truth</strong> (<em>Tuple</em><em>[</em><em>torch.Tensor</em><em>, </em><em>torch.Tensor</em><em>]</em>) – A tuple of two tensors. The
 first one has the shape (num_subsets, num_test_samples), which is the
 ground-truth target values for all test samples under <cite>num_subsets</cite> models,