Python: Advanced Guide to Artificial Intelligence

Naive Neural Network policy for Reinforcement Learning

We proceed with the policy as follows:

1. Let us implement a naive neural network-based policy. Define a new policy to use the neural network based predictions to return the actions:

def policy_naive_nn(nn,obs):    return np.argmax(nn.predict(np.array([obs])))

2. Define nn as a simple one layer MLP network that takes the observations having four dimensions as input, and produces the probabilities of the two actions:

from keras.models import Sequentialfrom keras.layers import Densemodel = Sequential()model.add(Dense(8,input_dim=4, activation='relu'))model.add(Dense(2, activation='softmax'))model.compile(loss='categorical_crossentropy',optimizer='adam')model.summary()

This is what the model ...