sekwiatkowski by komputation

Komputation

Komputation is a neural network framework for the JVM written in the Kotlin programming language.

Maven

Komputation is available through Maven Central:

<dependency>
    <groupId>com.komputation</groupId>
    <artifactId>komputation</artifactId>
    <version>0.11.3</version>
</dependency>

Layers

CPU demos

GPU/CUDA demos

Sample code

The following code instantiates a GPU-accelerated convolutional neural network for sentence classification:

    val network = Network(
        batchSize,
        lookupLayer(embeddings, maximumDocumentLength, hasVariableLength, embeddingDimension, optimization),
        convolutionalLayer(embeddingDimension, maximumDocumentLength, hasVariableLength, numberFilters, filterWidth, filterHeight, initialization, initialization, optimization),
        reluLayer(numberFilters),
        dropoutLayer(numberFilters, random, keepProbability),
        projectionLayer(numberFilters, numberCategories, initialization, initialization, optimization),
        softmaxLayer(numberCategories)
    )

See the TREC demo for more details.

Initialization

Loss functions

Optimization

compile "com.komputation:komputation:0.11.3"

Related Libraries

komputation

Komputation is a neural network framework for the JVM written in Kotlin and CUDA C.

Last updated 3 mins ago