From the course: Full-Stack Deep Learning with Python
Unlock this course with a free trial
Join today to access over 24,400 courses taught by industry experts.
Set up the dataset and data loader - Python Tutorial
From the course: Full-Stack Deep Learning with Python
Set up the dataset and data loader
- [Instructor] Now that we know how ML flow tracks runs within experiments, we are ready to train our dense neural network for image classification. Once again, A DNN is not the right model for this, but it'll give us a handle on how ML flow works and will work well as a preparatory step before convolutional neural network training. The first thing I'm going to to do is set up a dataset to access our emnistS data. A dataset is just an abstraction that represents your data and it typically provides access to individual data samples one at a time. This emnist dataset class inherits from the dataset base class in PyTorch. Within the init method, we initialize the images and labels as torch tensors. Observe that the init method takes in the data frame for our training or test data as an input argument. The Len Method returns the length of the dataset. That's just the length of the images get item allows us to access the data…
Practice while you learn with exercise files
Download the files the instructor uses to teach the course. Follow along and learn by watching, listening and practicing.
Contents
-
-
-
-
Loading and exploring the EMNIST dataset4m 53s
-
(Locked)
Logging metrics, parameters, and artifacts in MLflow6m 6s
-
(Locked)
Set up the dataset and data loader3m 47s
-
(Locked)
Configuring the image classification DNN model4m 56s
-
(Locked)
Training a model within an MLflow run4m 6s
-
(Locked)
Exploring parameters and metrics in MLflow4m 19s
-
(Locked)
Making predictions using MLflow artifacts5m 21s
-
-
-
-