@startuml

!include skinparams.puml

actor "Human user" as user

boundary "Training CLI" as cli
control "Read files" as read
control "Create initial model" as initial
entity "Neural network" as nn
control "Train network" as train
boundary "Model file" as model

user -> cli : provide input files
cli -> read
read -> initial
initial -> nn
loop until final epoch reached
    train <- nn : expose to training
    train -> nn : update
end
nn -> model : save neural network model

@enduml