LINUX.ORG.RU

Split on test and train sets

 


0

3

Всем добра!
Есть массив данных AxBxC и массив меток 1xC. Как в питоне образовать train и test наборы? Я пробовал с помощью cross_validation.train_test_split из sklearn. Да вот незадача, функция, как я понял, для тех случаев, когда вектор признаков для одной метки- это вектор, а не матрица (АхВ), как у меня. Мне б что нибудь, что просто сгенерирует индексы (как в R и Matlab), а не сами массивы дробить будет.

★★★

дык а в чем проблема сгенерировать индексы-то? создайте свой массив индексов, и его уже поделите случайным образом на train/validation/test/etc.

idx_train, idx_test = sklearn.cross_validation.train_test_split(
  numpy.linspace(0, 99, 100, dtype=int), train_size=0.7)

ну, или можно в чистом питоне

idx = range(0, 100)
random.shuffle(idx)
idx_train = idx[:70]
idx_test = idx[70:]

подойдет такое?

motus ()
Вы не можете добавлять комментарии в эту тему. Тема перемещена в архив.