def test_check_explainer_2(self): """ Unit test check explainer 2 """ df = pd.DataFrame(range(0, 21), columns=['id']) df['y'] = df['id'].apply(lambda x: 1 if x < 10 else 0) df['x1'] = np.random.randint(1, 123, df.shape[0]) df['x2'] = np.random.randint(1, 3, df.shape[0]) df = df.set_index('id') clf = cb.CatBoostClassifier(n_estimators=1).fit(df[['x1', 'x2']], df['y']) explainer = shap.TreeExplainer(clf) assert explainer.__class__.__base__.__name__ == "Explainer" check_explainer(explainer)
def check_explainer(self, explainer): """ Check if explainer class correspond to a shap explainer object """ return check_explainer(explainer)
def test_check_explainer_1(self): """ Unit test check explainer 1 """ explainer = None assert check_explainer(explainer) == None
def test_check_explainer_1(self): """ Unit test check explainer 1 """ explainer = None self.assertRaises(ValueError, check_explainer(explainer))