def test_estimators(self): ops = Ops.get_available_estimators() ops_names = [op.name() for op in ops] self.assertIn('LogisticRegression', ops_names) self.assertIn('MLPClassifier', ops_names) self.assertNotIn('PCA', ops_names)
def test_interpretable_estimators(self): ops = Ops.get_available_estimators({'interpretable'}) ops_names = [op.name() for op in ops] self.assertIn('KNeighborsClassifier', ops_names) self.assertNotIn('MLPClassifier', ops_names) self.assertNotIn('PCA', ops_names)
def test_interpretable_estimators(self): ops = Ops.get_available_estimators({"interpretable"}) ops_names = [op.name() for op in ops] self.assertIn("KNeighborsClassifier", ops_names) self.assertNotIn("MLPClassifier", ops_names) self.assertNotIn("PCA", ops_names)