diff --git a/extra_keras_datasets/emnist.py b/extra_keras_datasets/emnist.py index 4ed8718..cfcecb4 100644 --- a/extra_keras_datasets/emnist.py +++ b/extra_keras_datasets/emnist.py @@ -46,6 +46,11 @@ def load_data(path="emnist_matlab.npz", type="balanced"): Tuple of Numpy arrays: `(input_train, target_train), (input_test, target_test)`. """ + # Log about loading + logging.basicConfig(level=logging.INFO) + logging.info('Loading dataset = emnist') + + # Load data path = get_file( path, origin=("http://www.itl.nist.gov/iaui/vip/cs_links/EMNIST/" "matlab.zip") diff --git a/extra_keras_datasets/iris.py b/extra_keras_datasets/iris.py index 385a8bd..de98f65 100644 --- a/extra_keras_datasets/iris.py +++ b/extra_keras_datasets/iris.py @@ -46,6 +46,11 @@ def load_data(path="iris.npz", test_split=0.2): Target structure: 0 = iris setosa; 1 = iris versicolor; 2 = iris virginica. """ + # Log about loading + logging.basicConfig(level=logging.INFO) + logging.info('Loading dataset = iris') + + # Load data path = get_file( path, origin=("http://archive.ics.uci.edu/ml/machine-learning-databases/" diff --git a/extra_keras_datasets/kmnist.py b/extra_keras_datasets/kmnist.py index 3d01d6f..d86e0ba 100644 --- a/extra_keras_datasets/kmnist.py +++ b/extra_keras_datasets/kmnist.py @@ -42,6 +42,10 @@ def load_data(path="kmnist.npz", type="kmnist"): Tuple of Numpy arrays: `(input_train, target_train), (input_test, target_test)`. """ + # Log about loading + logging.basicConfig(level=logging.INFO) + logging.info('Loading dataset = kmnist') + # Load training images path_train = get_file( f"{path}_{type}_train_imgs", diff --git a/extra_keras_datasets/stl10.py b/extra_keras_datasets/stl10.py index 4a935bf..8146c1c 100644 --- a/extra_keras_datasets/stl10.py +++ b/extra_keras_datasets/stl10.py @@ -47,6 +47,10 @@ def load_data(path="stl10_matlab.tar.gz"): Tuple of Numpy arrays: `(input_train, target_train), (input_test, target_test)`. """ + # Log about loading + logging.basicConfig(level=logging.INFO) + logging.info('Loading dataset = stl-10') + # Set local .tar.gz path and delete if already available local_targz_path = "./stl-10" shutil.rmtree(local_targz_path, ignore_errors=True) diff --git a/extra_keras_datasets/svhn.py b/extra_keras_datasets/svhn.py index 68e6d54..6ba3431 100644 --- a/extra_keras_datasets/svhn.py +++ b/extra_keras_datasets/svhn.py @@ -48,6 +48,11 @@ def load_data(path="svhn_matlab.npz", type="normal"): Tuple of Numpy arrays: `(input_train, target_train), (input_test, target_test)`. """ + # Log about loading + logging.basicConfig(level=logging.INFO) + logging.info('Loading dataset = svhn') + + # Load data path_train = get_file( f"{path}_train", origin=("http://ufldl.stanford.edu/housenumbers/" "train_32x32.mat")