Skip to content

Commit edb7418

Browse files
interpret-mlnopdiveHarsha-Nori
committed
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>
1 parent 3eb19f3 commit edb7418

File tree

1 file changed

+2
-2
lines changed
  • python/interpret-core/interpret/glassbox/ebm/test

1 file changed

+2
-2
lines changed

python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ def test_unknown_multiclass_category():
4646
X_train['cat_feature'] = [np.random.choice(['a', 'b', 'c']) for x in range(X_train.shape[0])]
4747
X_test['cat_feature'] = ['d' for x in range(X_test.shape[0])] # Unknown category in test set
4848

49-
X_train['cat_feature'][1] = np.nan
50-
X_test['cat_feature'][1] = np.nan
49+
# X_train['cat_feature'][1] = np.nan
50+
# X_test['cat_feature'][1] = np.nan
5151

5252
clf = ExplainableBoostingClassifier()
5353
clf.fit(X_train, y_train)

0 commit comments

Comments
 (0)