risenW · July 14, 2020 08:33
diff --git a/split.py b/split.py
 from sklearn.model_selection import train_test_split
 Xtrain, Xtest = train_test_split(ratings_df, test_size=0.2, random_state=1)
 print(f"Shape of train data: {Xtrain.shape}")
 print(f"Shape of test data: {Xtest.shape}")
	from sklearn.model_selection import train_test_split
	Xtrain, Xtest = train_test_split(ratings_df, test_size=0.2, random_state=1)
	print(f"Shape of train data: {Xtrain.shape}")
	print(f"Shape of test data: {Xtest.shape}")