one_hot_encode.sql

{% macro one_hot_encode(column, source_relation='', source_condition='true', categories=[]) %}
    {{ return(adapter.dispatch('one_hot_encode', 'dbt_ml_inline_preprocessing')(column, source_relation, source_condition, categories)) }}
{% endmacro %}

{% macro default__one_hot_encode(column, source_relation, source_condition, categories)  %}

    {# Get the unique values that exist in the column to be encoded #}

    {% if categories == [] and source_relation == '' %}
        {% do exceptions.warn('Either source relation or categories must be set') %}
    {% elif source_relation == '' %}
        {% set category_values = categories %}
    {% else %}
        {% set category_values = dbt_utils.get_column_values(
            table=source_relation,
            column=column,
            order_by=column,
            where=
                column
                + ' is not null '
                + ' and '
                + source_condition
            ) or []    
        %}
    {% endif %}

    {# Generate a CASE statement for each unique value #}

    {% for category in category_values %}

        case
            when {{ column }} = '{{ category }}' then 1
            else 0
        end as is_{{ column }}__{{ dbt_utils.slugify(category | string) }},

    {% endfor %}

    case
        when {{ column }} is null then 1
        else 0
    end as is_{{ column }}__

{% endmacro %}