Written by Pedro Marquez who lives and works in Austin, Texas, doing all things front-end (and some back-end too).
Tag: pytorch
Create a ML classification model with PyTorch
October 03, 2022One 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.