Better organization of generated diagrams

Created by: igabriel85

Currently for advanced optimization during training the generated diagrams/plots are lumped together, should be separate folders.