405f7b2f686fb02713cede4fc1b03c249bf6898f,numerai/data.py,Data,cv,#Data#Any#Any#,70

Before Change

            dtrain = self[idx]
            idx = np.zeros(self.size, dtype=np.bool)
            for i in test_index:
                idx = np.logical_or(idx, era == eras[i])
            dtest = self[idx]
            yield dtrain, dtest

    def to_hdf(self, path_or_buf, **kwargs):

After Change

        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

Instances