Performing regression instead of categorical classification