Как разделить папку с несколькими наборами данных на поезд и протестировать с помощью PyTorch
У меня есть папка с 48 файлами сигналов ЭКГ. Файлы включают записи сигналов ЭКГ и аннотацию в форматах .dat и .atr. Я хочу разделить их, чтобы обучить и протестировать модель ИИ. Я буду использовать PyTorch и хочу узнать простой способ сделать это на Python. Я предпочитаю настраиваемое разделение с определенным количеством файлов, чтобы они находились в процессе обучения, а остальные — в тесте.
Например: Поезд: ['101', '104','107'] Тест: ['102', '105', '106']
Спасибо