mbohde
mbohde

Reputation: 139

Does it matter what dtype data is passed into a Tensorflow model for predictions?

So I have a TensorFlow model that I trained and it when I print the type of the data that I pass into the model to be trained, it is

<class 'tensorflow.python.data.ops.dataset_ops.MapDataset'>

Do I need to pass data of this same type into the model when I use .predict to make predictions from it or can I just pass it like a numpy array with the same dimensions?

Upvotes: 1

Views: 260

Answers (1)

Djib2011
Djib2011

Reputation: 7432

This just tells you that you trained your model with a tf.data.Dataset.

You can predict using a regular numpy array or a tensor, as long as it has the same input shape as the data you used to train the model on!

Upvotes: 1

Related Questions