Added DataSet Functions (aimacode#333)

antmarakis · norvig · commit 2de89e57fcdc · 2017-03-18T01:21:40.000-07:00
* Update learning.py

* Added remove_examples function
diff --git a/learning.py b/learning.py
@@ -154,6 +154,18 @@ def sanitize(self, example):
         return [attr_i if i in self.inputs else None
                 for i, attr_i in enumerate(example)]
 
+    def classes_to_numbers(self,classes=None):
+        """Converts class names to numbers."""
+        if not classes:
+            # If classes were not given, extract them from values
+            classes = sorted(self.values[self.target])
+        for item in self.examples:
+            item[self.target] = classes.index(item[self.target])
+            
+    def remove_examples(self,value=""):
+        """Remove examples that contain given value."""
+        self.examples = [x for x in self.examples if value not in x]
+
     def __repr__(self):
         return '<DataSet({}): {:d} examples, {:d} attributes>'.format(
             self.name, len(self.examples), len(self.attrs))
