LennartPurucker commited on
Commit
41ada02
·
1 Parent(s): a86996f

fix: add impute map for TB Lite

Browse files
Files changed (1) hide show
  1. main.py +4 -1
main.py CHANGED
@@ -342,7 +342,10 @@ def _get_lbs() -> tuple[LBContainer, ...]:
342
  file_name="lite/full-imputed/tabarena_leaderboard.csv",
343
  overview_image_name="lite/full-imputed/tuning-impact-elo.png",
344
  blurb="Leaderboard for one split (1st fold, 1st repeat) for all datasets including all (imputed) models.",
345
- imputed_map=None,
 
 
 
346
  )
347
  ta_clf = LBContainer(
348
  name="TabArena-Classification",
 
342
  file_name="lite/full-imputed/tabarena_leaderboard.csv",
343
  overview_image_name="lite/full-imputed/tuning-impact-elo.png",
344
  blurb="Leaderboard for one split (1st fold, 1st repeat) for all datasets including all (imputed) models.",
345
+ imputed_map={
346
+ "TabPFNv2": 35.29,
347
+ "TabICL": 29.41,
348
+ },
349
  )
350
  ta_clf = LBContainer(
351
  name="TabArena-Classification",