Set --balance to True by default

EleutherAI · Mar 22, 2023 · ae4efcf · ae4efcf
1 parent 1d60f5b
commit ae4efcf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/elk/extraction/prompt_dataset.py b/elk/extraction/prompt_dataset.py
@@ -65,7 +65,7 @@ class PromptConfig(Serializable):
     """
 
     dataset: str = field(positional=True)
-    balance: bool = False
+    balance: bool = True
     data_dir: Optional[str] = None
     label_column: Optional[str] = None
     num_classes: Optional[int] = None