Request for Modification: filter_triplets Function

[Juniork725]

I am proposing a modification to the filter_triplets function.

The proposed modification is to change:

    if min_sc > 0:
        itemcount = get_count(tp, 'movieId')
        tp = tp[tp['movieId'].isin(itemcount.index[itemcount >= min_sc])]
    if min_uc > 0:
        usercount = get_count(tp, 'userId')
        tp = tp[tp['userId'].isin(usercount.index[usercount >= min_uc])]

to:

    if min_sc > 0:
        itemcount = get_count(tp, 'movieId')
        tp = tp[tp['movieId'].isin(itemcount[itemcount >= min_sc]['movieId'])]  # Modified

    if min_uc > 0:
        usercount = get_count(tp, 'userId')
        tp = tp[tp['userId'].isin(usercount[usercount >= min_uc]['userId'])]  # Modified

The reason for this change is that 'movieId' and 'userId' values may not align perfectly with the index due to possible gaps or mismatches.

Thank you for considering this adjustment.

Best regards,
JunHyuck