From ba7ee783cb165dfa2ce8936af85e4e3310507eb1 Mon Sep 17 00:00:00 2001 From: Antonis Maronikolakis Date: Tue, 7 Mar 2017 23:45:39 +0200 Subject: [PATCH 1/2] Update learning.py --- learning.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/learning.py b/learning.py index df5d6fce3..5ad2d035d 100644 --- a/learning.py +++ b/learning.py @@ -152,6 +152,14 @@ 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 __repr__(self): return ''.format( self.name, len(self.examples), len(self.attrs)) From 2404824ce48991375b1cf1de7b4be2e9b517e34d Mon Sep 17 00:00:00 2001 From: Antonis Maronikolakis Date: Thu, 9 Mar 2017 14:33:45 +0200 Subject: [PATCH 2/2] Added remove_examples function --- learning.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/learning.py b/learning.py index 5ad2d035d..2b571240d 100644 --- a/learning.py +++ b/learning.py @@ -159,6 +159,10 @@ def classes_to_numbers(self,classes=None): 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(