My latest book "Backend Developer in 30 Days" is out! You can get it at Amazon
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.