We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0042079 commit 153ceccCopy full SHA for 153cecc
bencheval/bencheval/tabarena.py
@@ -1763,7 +1763,7 @@ def dataset_fold_similarity(
1763
if fold_sim.shape[0] >= 2:
1764
# consider only upper triangle (excluding diagonal)
1765
sim_vals = fold_sim.where(np.triu(np.ones(fold_sim.shape), k=1).astype(bool))
1766
- stacked = sim_vals.stack(dropna=True) # MultiIndex (seed_i, seed_j) -> similarity
+ stacked = sim_vals.stack(future_stack=True) # MultiIndex (seed_i, seed_j) -> similarity
1767
if len(stacked) > 0:
1768
(i_max, j_max), v_max = stacked.idxmax(), float(stacked.max())
1769
(i_min, j_min), v_min = stacked.idxmin(), float(stacked.min())
0 commit comments