add regression mode and classification mode