From edb7418cd12865e55e352ace6c61ff5cef8e1061 Mon Sep 17 00:00:00 2001 From: InterpretML <50031429+interpret-ml@users.noreply.github.com> Date: Wed, 13 Jan 2021 14:12:11 -0800 Subject: [PATCH] Update test_ebm.py Co-Authored-By: S Jenkins <69673492+nopdive@users.noreply.github.com> Co-Authored-By: Harsha-Nori <5302119+Harsha-Nori@users.noreply.github.com> --- python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py b/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py index 77f251a7d..364847c63 100644 --- a/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py +++ b/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py @@ -46,8 +46,8 @@ def test_unknown_multiclass_category(): X_train['cat_feature'] = [np.random.choice(['a', 'b', 'c']) for x in range(X_train.shape[0])] X_test['cat_feature'] = ['d' for x in range(X_test.shape[0])] # Unknown category in test set - X_train['cat_feature'][1] = np.nan - X_test['cat_feature'][1] = np.nan + # X_train['cat_feature'][1] = np.nan + # X_test['cat_feature'][1] = np.nan clf = ExplainableBoostingClassifier() clf.fit(X_train, y_train)