from surprise import SVD,Reader,Dataset
from surprise.model_selection import KFold
ratings =  pd.read_csv("/content/ratings_small.csv")
data = Dataset.load_from_df(ratings[['userId','movieId','rating']],reader)
kf = KFold(n_splits=5)
kf.split(data)