현재 쓰고 있는 stratify 기법

[MignonDeveloper]

def __make_stratified_key(self, df):
    stratified = df.copy()
    result = []

    prev_userID = stratified.userID[0]
    for userID, answerCode in zip(stratified.userID, stratified.answerCode.shift(1)):
        if prev_userID != userID:
            result.append(answerCode)
            prev_userID = userID
    
    result.append(stratified.answerCode.loc[stratified.shape[0] -1])
    self.user_stratified_key = result

마지막 문제를 맞췄는지 틀렸는지를 기준으로 stratify를 진행했습니다! seed는 1997 입니당