Bugfix in ablate_unfiltered

gaochonga · Nov 13, 2017 · 7cf2f91 · 7cf2f91
1 parent 6bcba14
commit 7cf2f91
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docqa/scripts/ablate_triviaqa_unfiltered.py b/docqa/scripts/ablate_triviaqa_unfiltered.py
@@ -13,7 +13,7 @@
 from docqa.scripts.ablate_triviaqa import get_model
 from docqa.text_preprocessor import WithIndicators
 from docqa.trainer import SerializableOptimizer, TrainParams
-from docqa.triviaqa.build_span_corpus import TriviaQaWebDataset
+from docqa.triviaqa.build_span_corpus import TriviaQaOpenDataset
 from docqa.triviaqa.training_data import ExtractMultiParagraphsPerQuestion
 
 
@@ -58,7 +58,7 @@ def main():
         test = RandomParagraphSetDatasetBuilder(120, "merge" if mode == "merge" else "group", True, oversample)
         train = StratifyParagraphSetsBuilder(30, mode == "merge", True, oversample)
 
-    data = TriviaQaWebDataset()
+    data = TriviaQaOpenDataset()
 
     params = TrainParams(
         SerializableOptimizer("Adadelta", dict(learning_rate=1)),