8.2 Do you wanna train a model?

TensorFlow is a machine learning framework developed at Google, and it is very widely used. You should refer to Appendix D, Installing the Tools, for installation instructions. Keep in mind that we will be using version 2.9.1. We will use TensorFlow in some of our quantum machine learning models, so it is a good idea to become familiar with it early on.

To keep things simple, we will tackle an artificial problem. We are going to prepare a dataset of elements belonging to one of two possible categories, and we will try to use machine learning to construct a classifier that can distinguish to which category any given input belongs.

Before we do anything, let us quickly import NumPy and set a seed:

import numpy as np 
 
seed = 128 
 
np.random.seed(seed)

We will later use this same seed with TensorFlow. And now, let’s generate the data!

Instead of generating a dataset by hand, we will use a function provided by the Python scikit-learn package (sklearn...