File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -154,6 +154,18 @@ def sanitize(self, example):
154154 return [attr_i if i in self .inputs else None
155155 for i , attr_i in enumerate (example )]
156156
157+ def classes_to_numbers (self ,classes = None ):
158+ """Converts class names to numbers."""
159+ if not classes :
160+ # If classes were not given, extract them from values
161+ classes = sorted (self .values [self .target ])
162+ for item in self .examples :
163+ item [self .target ] = classes .index (item [self .target ])
164+
165+ def remove_examples (self ,value = "" ):
166+ """Remove examples that contain given value."""
167+ self .examples = [x for x in self .examples if value not in x ]
168+
157169 def __repr__ (self ):
158170 return '<DataSet({}): {:d} examples, {:d} attributes>' .format (
159171 self .name , len (self .examples ), len (self .attrs ))
You can’t perform that action at this time.
0 commit comments