Minor changes

2020-10-22 14:49:27 +02:00
parent fb3edbbf64
commit 24778be577
1 changed files with 4 additions and 0 deletions
@@ -164,6 +164,10 @@ def main():
    test['features'], test['labels'] = read_mnist('t10k-images-idx3-ubyte.gz',
                                                  't10k-labels-idx1-ubyte.gz')

+    print(type(train))
+    print(type(train['features']))
+    print(type(train['labels']))
+
    # Step 3:
    # Explore the dataset
    print('Number of training images:', train['features'].shape[0])