Fixes parameter counts in tabtransformer.py

rama100 · web-flow · commit e9aa613a0d45 · 2025-03-13T14:29:14.000-04:00
diff --git 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.
 """
 
 """
