Enable passing a numpy.random.RandomState
object as random_state_model
in the train functions.
#3
Labels
numpy.random.RandomState
object as random_state_model
in the train functions.
#3
Currently, the train functions only accept integers as the model random state in
random_state_model
as it makes the internal handling of the differentRandomState
objects eventually required for different ranks in theDistributedRandomForest
class easier.For the data-related random state, each rank needs the same seed /
RandomState
object which makes it easy to further process aRandomState
object passed here. For the model, the original workflow took the model base seed, added the local rank, and turned it into a rank-specificRandomState
object which would not work the same if aRandomState
object would be passed here. If we would allow passing aRandomState
object for the model as well, it could be used to generate either the model base seed (in the case of local model seed = model base seed + rank) or the sequence ofcomm.size
integers used as local seeds.The text was updated successfully, but these errors were encountered: