training and test code for neonatal models

[cor2ni]

I've added the code for training and testing neonatal models including data preparation and augmentation. The test code can test combined and single models depending on a parameter.

[hongweilibran]

thanks!