diff --git a/learning.py b/learning.py index df5d6fce3..2b571240d 100644 --- a/learning.py +++ b/learning.py @@ -152,6 +152,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 ''.format( self.name, len(self.examples), len(self.attrs))