jeudi 3 décembre 2020

sklearn.model_selection.train_test_split random state

I am training a computer vision model. I divide the images in 3 datasets: training, validation and testing.

So that I get always the same images in training, vaidation and testing, I use the random_state parameter of train_test_split function.

However, I have a problem:

I am training and testing on two different computers (linux and windows). I thought that the results for a given random state would be same but they aren't.

Is there a way that I get the same results on both computers ?

I can't divide the images in 3 folders (training, validation and testing) since I want to change the test size and validation size during different experiments.




Aucun commentaire:

Enregistrer un commentaire