COSMIT Base class for KFold/StratifiedKFold validation