ap7sc/task5
2
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Make mnist downsizing easily configurable

Browse source
This commit is contained in:
Peter Vacho 2024-12-09 12:52:55 +01:00
parent e11750d54b
commit 8da828c817
Signed by: school
GPG key ID: 8CFC3837052871B4
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/__main__.py
View file

@ -16,6 +16,10 @@ from sklearn.model_selection import GridSearchCV, train_test_split
from sklearn.neighbors import KNeighborsClassifier
from sklearn.utils import Bunch
# Set to -1 to disable downsizing
DOWNSIZE_MNIST: int = 8000
# pyqt6 only bundles Windows & Fusion styles, which means that if you use a
# different preferred qt style, a warning would be produced. This gets rid
# of that warning and removes the env-var override.
@ -132,7 +136,10 @@ def main() -> None:
# Working with the entire dataset would be way too computationally expensive
# (TSNE would take hours, if not more), instead, downsize the dataset and work
# with a smaller sample
mnist = load_data().downsize(8000)
if DOWNSIZE_MNIST != -1:
mnist = load_data().downsize(DOWNSIZE_MNIST)
else:
mnist = load_data()
print()