Поделиться через


neuralNet: neuralNet

Создает список, содержащий имя функции и аргументы, для обучения модели NeuralNet с rxEnsemble.

Использование

  neuralNet(numHiddenNodes = 100, numIterations = 100, optimizer = sgd(),
    netDefinition = NULL, initWtsDiameter = 0.1, maxNorm = 0,
    acceleration = c("sse", "gpu"), miniBatchSize = 1, ...)
 

Аргументы

numHiddenNodes

Число скрытых узлов в нейронной сети по умолчанию. По умолчанию используется значение 100.

numIterations

Число итераций в полном наборе обучения. По умолчанию используется значение 100.

optimizer

Список, указывающий алгоритм оптимизации sgdили adaptive. Этот список можно создать с помощью sgd или adaDeltaSgd. Значение по умолчанию — sgd.

netDefinition

Определение структуры нейронной сети на языке Net#. Дополнительные сведения о языке Net# см. на странице Reference Guide

initWtsDiameter

Задает диаметр начальных весовых коэффициентов, который определяет диапазон, из которого берутся значения для первоначальных весовых коэффициентов обучения. Весовые коэффициенты инициализируются случайным образом из этого диапазона. Значение по умолчанию — 0,1.

maxNorm

Задает верхнюю границу для ограничения нормы входящего вектора весов в каждой скрытой единице. Это может быть важным в случае с максимальной выработкой нейронных сетей, а также в случаях, когда обучение приводит к появлению неограниченных весовых коэффициентов.

acceleration

Задает тип используемого аппаратного ускорения. Возможные значения: sse и gpu. Для ускорения GPU рекомендуется использовать параметр miniBatchSize со значением, превышающим единицу. Если вы хотите использовать ускорение GPU, необходимо выполнить дополнительные действия по настройке вручную.

  • Скачайте и установите NVidia CUDA Toolkit 6.5 (CUDA Toolkit).
  • Скачайте и установите библиотеку NVidia cuDNN версии 2 (cudnn Library).
  • Найдите каталог библиотек в пакете MicrosoftRML, вызвав system.file("mxLibs/x64", package = "MicrosoftML").
  • Скопируйте файлы cublas64_65.dll, cudart64_65.dll и cusparse64_65.dll из набора инструментов CUDA версии 6.5 в каталог библиотек пакета MicrosoftML.
  • Скопируйте файл cudnn64_65.dll из библиотеки cuDNN версии 2 в каталог библиотек пакета MicrosoftML.

miniBatchSize

Задает размер мини-пакета. Рекомендуемые значения — от 1 до 256. Этот параметр используется только в том случае, если для ускорения применяется GPU. При указании более высокого значения может увеличиться скорость обучения, но уменьшиться точность. Значение по умолчанию — 1.

...

Дополнительные аргументы.