Skip to content

Commit 94ad47e

Browse files
authored
Fixes parameter counts in tabtransformer.ipynb (#2116)
* Fixes parameter counts in tabtransformer.ipynb * Fixes parameter counts in tabtransformer.md * Updates parameter counts in tabtransformer.py
1 parent 4c7bcd3 commit 94ad47e

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

examples/structured_data/ipynb/tabtransformer.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -781,7 +781,7 @@
781781
"source": [
782782
"The TabTransformer model achieves ~85% validation accuracy.\n",
783783
"Note that, with the default parameter configurations, both the baseline and the TabTransformer\n",
784-
"have similar number of trainable weights: 109,629 and 92,151 respectively, and both use the same training hyperparameters."
784+
"have similar number of trainable weights: 109,895 and 87,745 respectively, and both use the same training hyperparameters."
785785
]
786786
},
787787
{
@@ -831,4 +831,4 @@
831831
},
832832
"nbformat": 4,
833833
"nbformat_minor": 0
834-
}
834+
}

examples/structured_data/md/tabtransformer.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -690,7 +690,7 @@ Validation accuracy: 84.59%
690690
</div>
691691
The TabTransformer model achieves ~85% validation accuracy.
692692
Note that, with the default parameter configurations, both the baseline and the TabTransformer
693-
have similar number of trainable weights: 109,629 and 92,151 respectively, and both use the same training hyperparameters.
693+
have similar number of trainable weights: 109,895 and 87,745 respectively, and both use the same training hyperparameters.
694694

695695
---
696696
## Conclusion

examples/structured_data/tabtransformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -553,7 +553,7 @@ def create_tabtransformer_classifier(
553553
"""
554554
The TabTransformer model achieves ~85% validation accuracy.
555555
Note that, with the default parameter configurations, both the baseline and the TabTransformer
556-
have similar number of trainable weights: 109,629 and 92,151 respectively, and both use the same training hyperparameters.
556+
have similar number of trainable weights: 109,895 and 87,745 respectively, and both use the same training hyperparameters.
557557
"""
558558

559559
"""

0 commit comments

Comments
 (0)