Merge pull request #132 from nicolewhite/tabular-args

Allow feature_names to be None

lime/lime_tabular.py

@@ -146,15 +146,9 @@ def __init__(self,
         """
         self.random_state = check_random_state(random_state)
         self.mode = mode
-        self.feature_names = list(feature_names)
-        self.categorical_names = categorical_names
-        self.categorical_features = categorical_features
-        if self.categorical_names is None:
-            self.categorical_names = {}
-        if self.categorical_features is None:
-            self.categorical_features = []
-        if self.feature_names is None:
-            self.feature_names = [str(i) for i in range(training_data.shape[1])]
+        self.categorical_names = categorical_names or {}
+        self.categorical_features = categorical_features or []
+        self.feature_names = feature_names or [str(i) for i in range(training_data.shape[1])]
 
         self.discretizer = None
         if discretize_continuous:

lime/tests/test_lime_tabular.py

@@ -535,6 +535,11 @@ def test_lime_tabular_explainer_not_equal_random_state(self):
 
         self.assertFalse(exp_1.as_map() != exp_2.as_map())
 
+    def testFeatureNames(self):
+        explainer = LimeTabularExplainer(training_data=np.array([[0., 1.], [1., 0.]]))
+
+        self.assertEqual(explainer.feature_names, ['0', '1'])
+
 
 if __name__ == '__main__':
     unittest.main()