Регистрация | Вход
with(NeuralNetwork(inputSize = 2, hiddenSize = 2, outputSize = 1)) { val trainingValues = listOf( NetworkData.create(listOf(0, 0), listOf(0)), NetworkData.create(listOf(0, 1), listOf(1)), NetworkData.create(listOf(1, 0), listOf(1)), NetworkData.create(listOf(1, 1), listOf(0))) train(trainingValues) test(trainingValues)}
val sentenceClassifier = cudaNetwork( batchSize, lookup(embeddings, maximumDocumentLength, embeddingDimension, optimization), convolution(numberFilters, filterWidth, filterHeight, initialization, optimization), relu(), dropout(random, keepProbability), dense(numberCategories, Activation.Softmax, initialization, optimization) )