fmda/reproducibility.py

   1 # make session reproducible
   2 import os
   3 environ={'TF_DETERMINISTIC_OPS':'1','PYTHONHASHSEED':'0','TF_CPP_MIN_LOG_LEVEL':'2'}
   4 # print('setting',environ)
   5 os.environ.update(environ)
   6 def set_seed(seed=123):
   7     import random
   8     random.seed(seed)
   9     import numpy as np
  10     np.random.seed(seed)
  11     import tensorflow as tf
  12     tf.random.set_seed(seed)
  13     tf.keras.utils.set_random_seed(seed)
  14     print('resetting random seeds to %i' % seed)
  15 # set_seed()
  16 # print('call set_seed() or set_seed(seed=value) to reset')
  17