improving diagnostics

[aaronspring]

• refactor existing: make more flexible for more metrics
• what else needed: support training validation with tensorboard?
• additional callbacks
• plot alpha_t(lag)