MultiDataSetIterator с INDArrays (не CSV-файлы) и несколькими выходами DL4J - PullRequest
0 голосов
/ 10 февраля 2020

Я хочу обучить ComputationGraph, который имеет два выхода ( эта модель ), и в моем скрипте у меня есть INDArrays (1 вход и 2 выхода), готовые для отправки в нейронную сеть, и кажется, что я следует использовать MultiDataSetIterator, чтобы иметь возможность настроить размер пакета перед использованием функции model.fit (). Я долго искал способ реализовать это, и я всегда находил ответы с помощью файлов CSV, но это не то, что я хочу использовать, потому что во время симуляции игры я создаю набор данных из INDArrays, которые сохраняются в память, и я не загружаю какой-либо CSV-файл.

Есть идеи, как создать мой MultiDataSetIterator для подачи моей функции fit ()?

1 Ответ

0 голосов
/ 10 февраля 2020

Вам не нужно использовать итератор нескольких наборов данных. Вы можете использовать мультиданные ( здесь ) или использовать массивы ndarrays ( здесь ), используя ваши ndarrays в памяти.

...