Proper ML training + class-conditional information + ML I/O