79c3d3004346ae19bb13332f84771a00a224e788,numerox/util.py,,cv,#Any#Any#Any#,5

Before Change

    kf = KFold(n_splits=kfold, shuffle=True, random_state=random_state)
    eras = self.era_dh.unique()
    for train_index, test_index in kf.split(eras):
        idx = self.df.era.isin(eras[train_index])
        dtrain = self[idx]
        idx = self.df.era.isin(eras[test_index])
        dtest = self[idx]
        yield dtrain, dtest

After Change

    eras = data.unique_era()
    for train_index, test_index in kf.split(eras):
        era_train = [eras[i] for i in train_index]
        era_test = [eras[i] for i in test_index]
        dtrain = data.era_isin(era_train)
        dtest = data.era_isin(era_test)
        yield dtrain, dtest

Instances