// Create a DQN solver (hidden layers have ReLu activation and output layer has softmax activation)
DQN dqn = new DQN.DQNBuilder()
		.discountFactor(0.995)
		.explorationRateDecay(0.995)
		.hiddenLayers(new int[] {30, 30, 30})
		.memorySize(100000)
		.miniBatchSize(32)
		.learningRate(0.01)
		.startingExplorationRate(1.0).build();

// Try to solve the gridworld using DQN (10000 epochs, max 200 steps per epoch)
dqn.run(new GridWorld(), 10000, 200);