def test_threshold_onehot_encoder():
st_helper = SklearnTestHelper()
tohe = ThresholdOneHotEncoder()
data = np.array([[10, 1, 7], [11, 3, 8], [11, 2, 9]], dtype=np.float32)
tohe.fit(data)
tohe.categories_ = [[10, 11], [1, 2, 3], [7, 8, 9]]
dshape = (relay.Any(), len(data[0]))
_test_model_impl(st_helper, tohe, dshape, data)
def test_threshold_encoder_with_a_column_under_threshold(
X, X_transformed_expected):
encoder = ThresholdOneHotEncoder(threshold=2)
encoder.fit(X)
assert_array_equal(encoder.transform(X).todense(), X_transformed_expected)
