@startuml

!include skinparams.puml

package "Engine module" {

    !include ImagoIO.pumlc
    !include GameEngine.pumlc
    !include DecisionAlgorithm.pumlc
    !include MCTS.pumlc
    !include MCTSNode.pumlc
    !include Keras.pumlc
    !include NeuralNetwork.pumlc
    !include DenseNeuralNetwork.pumlc
    !include ConvNeuralNetwork.pumlc

    ImagoIO ..> GameEngine
    GameEngine ..> DecisionAlgorithm

    DecisionAlgorithm <|.. MCTS
    MCTSNode <. MCTS
    MCTSNode -> MCTSNode
    MCTSNode o--> MCTSNode

    DecisionAlgorithm <|.. Keras
    Keras ..> NeuralNetwork
    NeuralNetwork <|-- DenseNeuralNetwork
    NeuralNetwork <|-- ConvNeuralNetwork

}

@enduml