Skip to content

Instantly share code, notes, and snippets.

View koljamaier's full-sized avatar

Kolja koljamaier

8 followers · 21 following
  • Hamburg, Germany
View GitHub Profile
@koljamaier
koljamaier / Multi-output model only evaluating as having one output column.
Created October 20, 2018 21:46 — forked from C4N4D4M4N/Multi-output model only evaluating as having one output column.
package com.IceKontroI.MODEL_TEST;
import org.deeplearning4j.nn.conf.ComputationGraphConfiguration;
import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
import org.deeplearning4j.nn.conf.inputs.InputType;
import org.deeplearning4j.nn.conf.layers.DenseLayer;
import org.deeplearning4j.nn.conf.layers.OutputLayer;
import org.deeplearning4j.nn.graph.ComputationGraph;
import org.nd4j.linalg.api.buffer.DataBuffer;
import org.nd4j.linalg.api.ndarray.INDArray;