miroesli / psscscs

Reinforcement Learning Battlesnake
MIT License
2 stars 1 forks source link

Store trained neural net in app/train.py in app/models/ #15

Closed AhmedNSidd closed 4 years ago

AhmedNSidd commented 4 years ago

In this issue, we need to store our trained, amazing neural network to disk in the app/models/ folder so we can actually use it later on. Everything to do this is more or less already set in place. When we run the train function for our alg, we'll get returned back a AlphaNNet object. These objects already have an instance function called save. The only thing that I think will need to be changed is the save function will need to specify the models folder. So the function will need to be changed from self.nnet.save(name + '.h5') to self.nnet.save('./models/' + name + '.h5') although not a 100% sure about this. Verification is left to the assignee.

AhmedNSidd commented 4 years ago

Addressed in Louis' #22 PR