Skip to content
Snippets Groups Projects
Commit 278917a4 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

SequentialExtractor: Apply extractor on training data always when apply=True

parent 04aace9a
No related branches found
No related tags found
1 merge request!112SequentialExtractor: Apply extractor on training data always when apply=True
......@@ -24,7 +24,13 @@ class MultipleExtractor(Extractor):
def train_one(self, e, training_data, extractor_file, apply=False):
if not e.requires_training:
return
if not apply:
return
if self.split_training_data_by_client:
training_data = [[e(d) for d in datalist]
for datalist in training_data]
else:
training_data = [e(d) for d in training_data]
# if any of the extractors require splitting the data, the
# split_training_data_by_client is True.
if e.split_training_data_by_client:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment