From 7b182e6f048d032645a967b4f0358616360901b3 Mon Sep 17 00:00:00 2001
From: luzhisheng <aiyingfeng110@qq.com>
Date: Thu, 20 Apr 2023 20:08:12 +0800
Subject: [PATCH] =?UTF-8?q?03=E6=95=B0=E6=8D=AE=E5=8A=A0=E8=BD=BD-?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E7=AD=89API=E4=BB=8B=E7=BB=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../03数据加载-模型等API介绍/MNIST模型初体验.py | 21 +++++++++++++++++++
 机器学习/03数据加载-模型等API介绍/__init__.py |  0
 2 files changed, 21 insertions(+)
 create mode 100644 机器学习/03数据加载-模型等API介绍/MNIST模型初体验.py
 create mode 100644 机器学习/03数据加载-模型等API介绍/__init__.py

diff --git a/机器学习/03数据加载-模型等API介绍/MNIST模型初体验.py b/机器学习/03数据加载-模型等API介绍/MNIST模型初体验.py
new file mode 100644
index 0000000..47528e5
--- /dev/null
+++ b/机器学习/03数据加载-模型等API介绍/MNIST模型初体验.py
@@ -0,0 +1,21 @@
+from torchvision.datasets import MNIST
+import matplotlib.pyplot as plt
+
+mnist_train = MNIST(root="./MNIST_data", train=True, download=True, transform=None)
+
+# 第一次运行会下载数据集
+print(mnist_train)
+
+# 数据集总量
+print(len(mnist_train))
+
+# 数据集元祖，（图片，标注）
+print(mnist_train[5000])
+image = mnist_train[5000][0]
+
+# 打印图片
+plt.imshow(image)
+plt.show()
+
+# 获取标注值
+print(mnist_train[5000][1])
diff --git a/机器学习/03数据加载-模型等API介绍/__init__.py b/机器学习/03数据加载-模型等API介绍/__init__.py
new file mode 100644
index 0000000..e69de29