def METHOD_NAME(self, df, params=None):
    """
    Transforms the input dataset with prediction columns representing model predictions.
    Prediction column names default to <label_column>__output. Override column names
    by calling `transformer.setOutputCols(col_names)`.
    Args:
        df: Input dataset, which is an instance of :py:class:`pyspark.sql.DataFrame`.
        params: An optional param map that overrides embedded params.
    Returns:
        Transformed dataset.
    """
    return super(HorovodModel, self).METHOD_NAME(df, params)