[Numpy-discussion] random.RandomState and deepcopy
Neal Becker
ndbecker2 at gmail.com
Fri Mar 13 13:34:06 EDT 2015
It is common that to guarantee good statistical independence between various
random generators, a singleton instance of an RNG is shared between them.
So I typically have various random generator objects, which (sometimes
several levels objects deep) embed an instance of RandomState.
Now I have a requirement to copy a generator object (without knowing exactly
what that generator object is).
My solution is to use deepcopy on the top-level object. But I need to
overload __deepcopy__ on the singleton RandomState object.
Unfortunately, RandomState doesn't allow customization of __deepcopy__ (or
anything else). And it has no __dict__.
My solution is:
class shared_random_state (object):
def __init__ (self, rs):
self.rs = rs
def __getattr__ (self, attr):
return getattr (self.rs, attr)
def __deepcopy__ (self):
return self
An example usage:
rs = shared_random_state (RandomState(0))
from exponential import exponential
e = exponential (rs, 1)
where exponential is:
class exponential (object):
def __init__ (self, rs, mu):
self.rs = rs
self.mu = mu
def __call__ (self, size=None):
if size == None:
return self.rs.exponential (self.mu, 1)[0]
else:
return self.rs.exponential (self.mu, size)
def __repr__ (self):
return 'exp(%s)' % self.mu
I wonder if anyone has any other suggestions? Personally, I would prefer if
numpy provided a more direct solution to this. Either by providing for
overloading RandomState deepcopy, or by making the copy behavior switchable
with a flag to the constructor.
--
Those who fail to understand recursion are doomed to repeat it
More information about the NumPy-Discussion
mailing list