Merge branch 'feature/advanced-configuration' of https://github.com/l…

…uxonis/datadreamer into feature/advanced-configuration
luxonis · Mar 25, 2024 · b4f8a16 · b4f8a16
2 parents 52f6426 + 6bd4642
commit b4f8a16
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/datadreamer/prompt_generation/lm_synonym_generator.py b/datadreamer/prompt_generation/lm_synonym_generator.py
@@ -34,7 +34,7 @@ class LMSynonymGenerator(SynonymGenerator):
 
     def __init__(
         self,
-        synonyms_number: int = 5,
+        synonyms_number: int = 3,
         seed: Optional[float] = 42,
         device: str = "cuda",
     ) -> None:

diff --git a/datadreamer/prompt_generation/synonym_generator.py b/datadreamer/prompt_generation/synonym_generator.py
@@ -29,7 +29,7 @@ class SynonymGenerator(ABC):
 
     def __init__(
         self,
-        synonyms_number: int = 5,
+        synonyms_number: int = 3,
         seed: Optional[float] = 42,
         device: str = "cuda",
     ) -> None:

diff --git a/datadreamer/prompt_generation/wordnet_synonym_generator.py b/datadreamer/prompt_generation/wordnet_synonym_generator.py
@@ -25,7 +25,7 @@ class WordNetSynonymGenerator(SynonymGenerator):
 
     def __init__(
         self,
-        synonyms_number: int = 5,
+        synonyms_number: int = 3,
         seed: Optional[float] = 42,
         device: str = "cuda",
     ) -> None:

diff --git a/media/coverage_badge.svg b/media/coverage_badge.svg