tf.keras.utils.set_random_seed  |  TensorFlow v2.16.1 (original) (raw)

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

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.datapipelines (which feature random shuffling). Likewise many workflows might leverage NumPy APIs.

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.