tf.keras.utils.set_random_seed
Stay organized with collections
Save and categorize content based on your preferences.
Sets all random seeds (Python, NumPy, and backend framework, e.g. TF).
tf.keras.utils.set_random_seed(
seed
)
Used in the notebooks
Used in the guide |
Used in the tutorials |
|
|
You can use this utility to make almost any Keras program fully
deterministic. Some limitations apply in cases where network communications
are involved (e.g. parameter server distribution), which creates additional
sources of randomness, or when certain non-deterministic cuDNN ops are
involved.
Calling this utility is equivalent to the following:
import random
random.seed(seed)
import numpy as np
np.random.seed(seed)
import tensorflow as tf # Only if TF is installed
tf.random.set_seed(seed)
import torch # Only if the backend is 'torch'
torch.manual_seed(seed)
Note that the TensorFlow seed is set even if you're not using TensorFlow
as your backend framework, since many workflows leverage tf.data
pipelines (which feature random shuffling). Likewise many workflows
might leverage NumPy APIs.
Arguments |
seed
|
Integer, the random seed to use.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-06-07 UTC.
[null,null,["Last updated 2024-06-07 UTC."],[],[],null,["# tf.keras.utils.set_random_seed\n\n\u003cbr /\u003e\n\n|---------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/keras-team/keras/tree/v3.3.3/keras/src/utils/rng_utils.py#L10-L56) |\n\nSets all random seeds (Python, NumPy, and backend framework, e.g. TF). \n\n tf.keras.utils.set_random_seed(\n seed\n )\n\n### Used in the notebooks\n\n| Used in the guide | Used in the tutorials |\n|-----------------------------------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------|\n| - [Use TF1.x models in TF2 workflows](https://www.tensorflow.org/guide/migrate/model_mapping) | - [Using DTensors with Keras](https://www.tensorflow.org/tutorials/distribute/dtensor_keras_tutorial) |\n\nYou can use this utility to make almost any Keras program fully\ndeterministic. Some limitations apply in cases where network communications\nare involved (e.g. parameter server distribution), which creates additional\nsources of randomness, or when certain non-deterministic cuDNN ops are\ninvolved.\n\nCalling this utility is equivalent to the following: \n\n import random\n random.seed(seed)\n\n import numpy as np\n np.random.seed(seed)\n\n import tensorflow as tf # Only if TF is installed\n tf.random.set_seed(seed)\n\n import torch # Only if the backend is 'torch'\n torch.manual_seed(seed)\n\nNote that the TensorFlow seed is set even if you're not using TensorFlow\nas your backend framework, since many workflows leverage [`tf.data`](../../../tf/data)\npipelines (which feature random shuffling). Likewise many workflows\nmight leverage NumPy APIs.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Arguments --------- ||\n|--------|----------------------------------|\n| `seed` | Integer, the random seed to use. |\n\n\u003cbr /\u003e"]]