Learning Based Agent Diagram