def test_classification_ignore_format(df_enc_numeric):
encoder = DecisionTreeEncoder(regression=False, ignore_format=True)
encoder.fit(df_enc_numeric[["var_A", "var_B"]], df_enc_numeric["target"])
X = encoder.transform(df_enc_numeric[["var_A", "var_B"]])
transf_df = df_enc_numeric.copy()
transf_df["var_A"] = [0.25] * 16 + [0.5] * 4 # Tree: var_A <= 1.5 -> 0.25 else 0.5
transf_df["var_B"] = [0.2] * 10 + [0.4] * 10 # Tree: var_B <= 0.5 -> 0.2 else 0.4
pd.testing.assert_frame_equal(X, transf_df[["var_A", "var_B"]])
def test_variables_cast_as_category(df_enc_category_dtypes):
df = df_enc_category_dtypes.copy()
encoder = DecisionTreeEncoder(regression=False)
encoder.fit(df[["var_A", "var_B"]], df["target"])
X = encoder.transform(df[["var_A", "var_B"]])
transf_df = df.copy()
transf_df["var_A"] = [0.25] * 16 + [0.5] * 4 # Tree: var_A <= 1.5 -> 0.25 else 0.5
transf_df["var_B"] = [0.2] * 10 + [0.4] * 10 # Tree: var_B <= 0.5 -> 0.2 else 0.4
pd.testing.assert_frame_equal(X, transf_df[["var_A", "var_B"]], check_dtype=False)
assert X["var_A"].dtypes == float
def test_regression(df_enc, df_enc_na):
random = np.random.RandomState(42)
y = random.normal(0, 0.1, len(df_enc))
encoder = DecisionTreeEncoder(regression=True, random_state=random)
encoder.fit(df_enc[["var_A", "var_B"]], y)
X = encoder.transform(df_enc[["var_A", "var_B"]])
transf_df = df_enc.copy()
transf_df["var_A"] = ([0.034348] * 6 + [-0.024679] * 10 + [-0.075473] * 4
) # Tree: var_A <= 1.5 -> 0.25 else 0.5
transf_df["var_B"] = [0.044806] * 10 + [-0.079066] * 10
pd.testing.assert_frame_equal(X.round(6), transf_df[["var_A", "var_B"]])
def test_transform_raises_error_if_df_contains_na(df_enc, df_enc_na):
# test case 4: when dataset contains na, transform method
with pytest.raises(ValueError):
encoder = DecisionTreeEncoder()
encoder.fit(df_enc_na[["var_A", "var_B"]], df_enc_na["target"])
encoder.transform(df_enc_na)
def test_non_fitted_error(df_enc):
with pytest.raises(NotFittedError):
encoder = DecisionTreeEncoder()
encoder.transform(df_enc)