diff --git a/pyfm/pylibfm.py b/pyfm/pylibfm.py index b0b71a4..5926106 100644 --- a/pyfm/pylibfm.py +++ b/pyfm/pylibfm.py @@ -174,7 +174,7 @@ def fit(self, X, y): if self.verbose == True: print("Creating validation dataset of %.2f of training for adaptive regularization" % self.validation_size) X_train, validation, train_labels, validation_labels = train_test_split( - X, y, test_size=self.validation_size) + X, y, test_size=self.validation_size, random_state=self.seed) self.num_attribute = X_train.shape[1] # Convert datasets to sklearn sequential datasets for fast traversal