simple-data-science-notebook/src/target_transformers.py

from typing import List
import numpy as np
import pandas as pd
from src.categorical_encoders import *
from src.numerical_transformers import *

def target_transformer(
    df: pd.DataFrame,
    target: str,
) -> pd.DataFrame:
    
    if df[target].dtypes[0] == "category":
        
        df = ordinal_encoder(df, target)
        
    elif df[target].dtypes[0] == "float":
        
        df = variable_transformer(df, target, variable_type="power_transformer")
        
    else:
        raise ValueError('target_type must be either "category" or "float"')
        
    return df