HMM data in folds train+test script