Compartir a través de

neuralNet: neuralNet

  • Artículo

Crea una lista que contiene el nombre de la función y los argumentos para entrenar un modelo NeuralNet con rxEnsemble.

Uso

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

Argumentos

numHiddenNodes

Número predeterminado de nodos ocultos de la red neuronal. El valor predeterminado es 100.

numIterations

Número de iteraciones del conjunto de entrenamiento completo. El valor predeterminado es 100.

optimizer

Lista que especifica el algoritmo de optimización sgd o adaptive. Esta lista se puede crear mediante sgd o adaDeltaSgd. El valor predeterminado es sgd.

netDefinition

Definición de Net# de la estructura de la red neuronal. Para más información sobre el lenguaje Net#, consulte Reference Guide.

initWtsDiameter

Establece el diámetro de pesos inicial que especifica el intervalo desde el que se dibujan los valores para los pesos de aprendizaje iniciales. Los pesos se inicializan aleatoriamente desde este intervalo. El valor predeterminado es 0.1.

maxNorm

Especifica un límite superior para restringir la norma del vector de peso entrante en cada unidad oculta. Esto puede resultar importante en el número máximo de redes neuronales y en casos en los que el entrenamiento genera pesos sin límite.

acceleration

Especifica el tipo de aceleración de hardware que se usará. Los valores posibles son "sse" y "gpu". En el caso de la aceleración de GPU, se recomienda usar un miniBatchSize mayor que uno. Si desea usar la aceleración de GPU, se requieren pasos de configuración manuales adicionales:

  • Descargue e instale NVidia CUDA Toolkit 6.5 (CUDA Toolkit).
  • Descargue e instale la biblioteca NVidia cuDNN v2 (cudnn Library).
  • Busque el directorio de bibliotecas del paquete MicrosoftRML llamando a system.file("mxLibs/x64", package = "MicrosoftML").
  • Copie cublas64_65.dll, cudart64_65.dll y cusparse64_65.dll de CUDA Toolkit 6.5 en el directorio de bibliotecas del paquete MicrosoftML.
  • Copie cudnn64_65.dll de la biblioteca cuDNN v2 en el directorio de bibliotecas del paquete MicrosoftML.

miniBatchSize

Establece el tamaño mínimo del lote. Los valores recomendados están comprendidos entre 1 y 256. Este parámetro solo se usa cuando la aceleración es GPU. El establecimiento de este parámetro en un valor mayor mejora la velocidad de entrenamiento, pero podría afectar negativamente a la precisión. El valor predeterminado es 1.

...

Argumentos adicionales.