Training dataset generation and training code upgrades

To make DAN training easier, we first need to improve the codebase.

• Import and format data from Arkindex
• Update training code