@startuml

class NeuralNetwork {
    int boardSize
    string path

    trainModel(games)
    saveModel(modelPath)
    pickMove(gameMove, player)
    saveModelPlot(path)
}

@enduml