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)