From 868f884bb4b255d5ed43b42eb4787beba9bee641 Mon Sep 17 00:00:00 2001 From: Marko Lalovic Date: Tue, 9 Jul 2024 20:20:04 +0200 Subject: [PATCH] prepares n_samples = 10000 samples to data --- scripts/prepare_data.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/scripts/prepare_data.py b/scripts/prepare_data.py index 38c37b9..f1f3ced 100755 --- a/scripts/prepare_data.py +++ b/scripts/prepare_data.py @@ -1,15 +1,13 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- -''' prepare_data.py: Saves n_samples images of digits to numpy array.''' +''' prepare_data.py: Saves n_samples = 10000 images of digits to numpy array.''' from __future__ import print_function # if you are using Python 2 import numpy as np from sklearn.datasets import fetch_openml import sys -# better use: n_samples = 10000 -n_samples = 1000 - +n_samples = 10000 if len(sys.argv) == 2: print('Setting n_samples to: %i' % (n_samples)) n_samples = int(sys.argv[1])