5,321
edits
TensorFlow: Difference between revisions
→Basics
(→Basics) |
|||
| Line 24: | Line 24: | ||
* Define a model, typically using [https://www.tensorflow.org/api_docs/python/tf/keras/Sequential tf.keras.Sequential] | * Define a model, typically using [https://www.tensorflow.org/api_docs/python/tf/keras/Sequential tf.keras.Sequential] | ||
* Call [https://www.tensorflow.org/api_docs/python/tf/keras/Model#compile <code>model.compile</code>] | * Call [https://www.tensorflow.org/api_docs/python/tf/keras/Model#compile <code>model.compile</code>] | ||
** Here you pass in your optimizer, loss function, metrics | ** Here you pass in your optimizer, loss function, and metrics. | ||
* Train your model by calling <code>model.fit</code> | * Train your model by calling <code>model.fit</code> | ||
** Here you pass in your training data | ** Here you pass in your training data, batch size, number of epochs, and training callbacks | ||
** For more information about callbacks, see [https://www.tensorflow.org/guide/keras/custom_callback Keras custom callbacks]. | |||
After training, you can use your model by calling <code>model.evaluate</code> | After training, you can use your model by calling <code>model.evaluate</code> | ||
===Custom Models=== | ===Custom Models=== | ||