predictoutput.jl

##### Beginning of file

import DataFrames

"""
"""
struct ImmutablePredictionsSingleLabelInt2StringTransformer <:
        AbstractEstimator
    index::T1 where T1 <: Integer
    levels::T2 where T2 <: AbstractVector
end

"""
"""
function set_feature_contrasts!(
        x::ImmutablePredictionsSingleLabelInt2StringTransformer,
        feature_contrasts::AbstractFeatureContrasts,
        )
    return nothing
end

"""
"""
function get_underlying(
        x::ImmutablePredictionsSingleLabelInt2StringTransformer;
        saving::Bool = false,
        loading::Bool = false,
        )
    return nothing
end

"""
"""
function get_history(
        x::ImmutablePredictionsSingleLabelInt2StringTransformer;
        saving::Bool = false,
        loading::Bool = false,
        )
    return nothing
end

"""
"""
function parse_functions!(
        transformer::ImmutablePredictionsSingleLabelInt2StringTransformer,
        )
    return nothing
end

"""
"""
function fit!(
        transformer::ImmutablePredictionsSingleLabelInt2StringTransformer,
        varargs...;
        kwargs...
        )
    if length(varargs) == 1
        return varargs[1]
    else
        return varargs
    end
end

"""
"""
function predict(
        transformer::ImmutablePredictionsSingleLabelInt2StringTransformer,
        single_labelpredictions::AbstractVector;
        kwargs...
        )
    single_labelpredictions = parse.(Int, single_labelpredictions)
    labelint2stringmap = _getlabelint2stringmap(
        transformer.levels,
        transformer.index,
        )
    result = Vector{String}(length(single_labelpredictions))
    for i = 1:length(result)
        result[i] = labelint2stringmap[single_labelpredictions[i]]
    end
    return result
end

"""
"""
function predict(
        transformer::ImmutablePredictionsSingleLabelInt2StringTransformer,
        single_labelpredictions::DataFrames.AbstractDataFrame;
        kwargs...
        )
    label_names = DataFrames.names(single_labelpredictions)
    result = DataFrames.DataFrame()
    for i = 1:length(label_names)
        result[label_names[i]] = predict(
            transformer,
            single_labelpredictions[label_names[i]];
            kwargs...
            )
    end
    return result
end

"""
"""
function predict_proba(
        transformer::ImmutablePredictionsSingleLabelInt2StringTransformer,
        varargs...;
        kwargs...
        )
    if length(varargs) == 1
        return varargs[1]
    else
        return varargs
    end
end

##### End of file