code: def predict(self, dataset, output_type='class', missing_value_action='auto'):
          _check_categorical_option_type('output_type', output_type,
                  ['class', 'margin', 'probability', 'probability_vector'])
          return super(_Classifier, self).predict(dataset,
                                                  output_type=output_type,
                                                  missing_value_action=missing_value_action)