diff options
Diffstat (limited to 'doc/diagrams/NeuralNetwork.pumlc')
| -rw-r--r-- | doc/diagrams/NeuralNetwork.pumlc | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/doc/diagrams/NeuralNetwork.pumlc b/doc/diagrams/NeuralNetwork.pumlc index 30f783e..f6c0e1c 100644 --- a/doc/diagrams/NeuralNetwork.pumlc +++ b/doc/diagrams/NeuralNetwork.pumlc @@ -1,8 +1,13 @@ @startuml class NeuralNetwork { - load(self, path) - save(self, path) + int boardSize + string path + + trainModel(games) + saveModel(modelPath) + pickMove(gameMove, player) + saveModelPlot(path) } @enduml |