FilesExpand file tree

 master

/

ExampleEmbeddedReberGrammar.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

34 lines (25 loc) · 855 Bytes

 master

/

ExampleEmbeddedReberGrammar.java

Copy path

Top

File metadata and controls

• Code
• Blame

34 lines (25 loc) · 855 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

import java.util.Random;

import model.Model;

import trainer.Trainer;

import util.NeuralNetworkHelper;

import datasets.EmbeddedReberGrammar;

import datastructs.DataSet;

public class ExampleEmbeddedReberGrammar {

public static void main(String[] args) throws Exception {

Random rng = new Random();

DataSet data = new EmbeddedReberGrammar(rng);

int hiddenDimension = 12;

int hiddenLayers = 1;

double learningRate = 0.001;

double initParamsStdDev = 0.08;

Model nn = NeuralNetworkHelper.makeLstm(

data.inputDimension,

hiddenDimension, hiddenLayers,

data.outputDimension, data.getModelOutputUnitToUse(),

initParamsStdDev, rng);

int reportEveryNthEpoch = 10;

int trainingEpochs = 1000;

Trainer.train(trainingEpochs, learningRate, nn, data, reportEveryNthEpoch, rng);

System.out.println("done.");

}

}