From e9aa613a0d45b86503809bc2de95b7b0543c71da Mon Sep 17 00:00:00 2001 From: Rama Ramakrishnan Date: Thu, 13 Mar 2025 14:29:14 -0400 Subject: [PATCH] Fixes parameter counts in tabtransformer.py --- examples/structured_data/tabtransformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/structured_data/tabtransformer.py b/examples/structured_data/tabtransformer.py index f8f3a808b8..304bc44259 100644 --- a/examples/structured_data/tabtransformer.py +++ b/examples/structured_data/tabtransformer.py @@ -553,7 +553,7 @@ def create_tabtransformer_classifier( """ The TabTransformer model achieves ~85% validation accuracy. Note that, with the default parameter configurations, both the baseline and the TabTransformer -have similar number of trainable weights: 109,629 and 92,151 respectively, and both use the same training hyperparameters. +have similar number of trainable weights: 109,895 and 87,745 respectively, and both use the same training hyperparameters. """ """