Example #1
0
    # "John ate Pizza with Jane.",
    # "Jane ate Waffles with Bob."
    # "John ate Cheese with Dog."
    # "Jane ate Pie with Bob."
]

rnnHand = RNNDataHandle(textData)
inputSize = rnnHand.buildVocab()
inputs, targets = rnnHand.buildInputTarget()
print(inputs)
print(targets)

hiddenPoss = int(inputSize * 0.75)

net = RNN(learningRate=0.01, hiddenSize=hiddenPoss, decayRate=0.1)
net.addLayer(inputLayer=True, inputSize=inputSize)
net.addLayer()
net.addLayer()
net.addLayer()
# net.addLayer()
net.addLayer(outputLayer=True, outputSize=inputSize)
net.trainNetwork(inputs, targets, epochs=31)
net.saveNeuralNet("rnn_net")

lastOut = []


# Makes prediciton for next word
def makePrediction(wordInput):
    return net.predict(wordInput)