fmussari

# Inspired by:  
# https://github.com/microsoft/recommenders/blob/main/recommenders/datasets/python_splitters.py
# - In "recommenders" library, when spliting by 'item', `python_stratified_split` puts both rows of movies with only two ratings in the training set.
# - This version here puts one in the training and one in validation set.

import pandas as pd
import numpy as np

DEFAULT_USER_COL = 'user'
DEFAULT_ITEM_COL = 'title'

fmussari