My latest book "Backend Developer in 30 Days" is out! You can get it at Amazon
← Back to all posts

Pedro's Tech Blog

Pedro Marquez

Written by Pedro Marquez who lives and works in Austin, Texas, doing all things front-end (and some back-end too).

Tag: pytorch

A hero image for this post about state management

Create a ML classification model with PyTorch

October 03, 2022

One of the most common tasks in Machine Learning is classification: Creating a model that, after being trained with a dataset, it can label specific examples of data into one or more categories.

In this post, we will use PyTorch -one of the most popular ML tools- to create and train a simple classification model using neural networks.