Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- static void main(String[] args) {
- // def net = createNetwork()
- def net = loadNetwork()
- INDArray features = Nd4j.create([1, 3, 1] as int[], (char) 'f')
- features.putScalar([0, 0, 0] as int[], 30807)
- features.putScalar([0, 1, 0] as int[], 726207)
- features.putScalar([0, 2, 0] as int[], 238.31)
- def labels = Nd4j.create([1, 1, 1] as int[], (char) 'f')
- labels.putScalar([0, 0, 0] as int[], 238.31)
- def dataSet = new DataSet(features, labels)
- normalizer.fitLabel(true)
- normalizer.fit(dataSet)
- normalizer.preProcess(dataSet)
- def result = net.output(features)
- normalizer.revertLabels(result)
- println result
- }
Add Comment
Please, Sign In to add comment