人工智能|机器学习开篇——从源代码开始认识机器学习
人工智能|机器学习开篇——从源代码开始认识机器学习 选择深度学习框架 对于初学者来说,当他们初次接触深度学习时,通常首要问题是如何选择合适的深度学习框架。截止2023年,推荐的框架有: Tensorflow Keras PyTorch MxNet Chainer Caffe Theano Deeplearning4j CNTK Torch 我也是个初学者,这里就不过多介绍框架的异同,若有兴趣可以点击查看详细介绍。在这篇文章里我们选择PyTorch作为深度学习框架。 使用PyTorch构建模型 准备数据 在机器学习中,算法和数据都是同等重要的,而获取、处理数据的过程甚至都是困难的。在本例中使用已收集且已特征化的数据来完成实例。 PyTorch加载数据使用两个抽象类型torch.utils.data.DataLoader和torch.utils.data.Dataset。 import torch from torch import nn from torch.utils.data import DataLoader from torchvision import datasets from torchvision.transforms import ToTensor PyTorch提供TorchText、TorchVision、TorchAudio三个领域模型来处理文本、图片、声音。本实例使用TorchVision来处理图片。 # Download training data from open datasets. training_data = datasets.FashionMNIST( root="data", train=True, download=True, transform=ToTensor(), ) # Download test data from open datasets. test_data = datasets.FashionMNIST( root="data", train=False, download=True, transform=ToTensor(), ) 输出: Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-images-idx3-ubyte.gz Downloading http://fashion-mnist....