ENH: better use of compcor train / test application